要注意,因为mmdeploy要用nvcc编译,所以本机的cuda toolkit也要安装,我因为只在conda环境装了cuda toolkit而忽视了,nvcc编译也能正常通过,只是运行时在trt nms那里会报错。 permuteData.cu文件我改了很多,终于调试出是sm_86的问题(我用的是3090,你的显卡不一样的话要查看主机cuda版本是否支持). 这里吐槽一下我之...
实验结果表明,我们的方法在标准 setting 下,不仅超越了 CondInst,SOLOv2 等单阶段模型,也超越了 Cascade Mask R-CNN 这样的多阶段模型。在使用与 RTMDet 一致的模型结构以及训练策略后,得到的 RTMDet-Ins 模型,精度不仅大幅超越 YOLOv5-seg,也超越了前几天刚刚推出的 YOLOv8,取得了实时实例分割的 SOTA。 表11....
在尝试使用mmdeploy部署rtmdet ins进行实例分割时，发现网上资源较少，github的问题也有不少未解决，只得自己动手，留作档案。构建时需严格遵循源代码构建流程，从下载到构建，避免第三方库问题。ppl.cv不支持cuda12，构建时需在cuda.cmake中设置与自己显卡对应的flag。使用了最新版的tensorrt和cudnn，版本...
图8. RTMDet-Ins 实例分割分支示意图为了最大化利用 mask 的标注信息,我们还将标签分配中的中心先验代价的检测框中心修改为了 mask 的重心。为了与其他实例分割方法做公平对比,我们使用 ResNet50 FPN 12 epoch 的标准 setting 进行了实验,同时参考 CondInst 加入了语义分割的辅助分支加快收敛。 实验结果表明,我们的...
Checklist I have searched related issues but cannot get the expected help. 2. I have read the FAQ documentation but cannot get the expected help. 3. The bug has not been fixed in the latest version. Describe the bug I want to parse the R...
RTMDet-Ins ONNX and TensorRT support. Modification Add rewriter for RTMDetInsHead Fix the rescale of the object detection model class(need to check other models) Inference Results(TRT) Please use configs/mmdet/instance-seg/instance-seg_rtmdet-ins_tensorrt_static-640x640.py mask AP is aligned...
checkpoint_file ='checkpoints/rtmdet-ins_l_8xb32-300e_coco_20221124_103237-78d1d652.pth' register_all_modules() model = init_detector(config_file, checkpoint_file, device='cuda:0') image = mmcv.imread('demo/demo.jpg',channel_order='rg...
具体来说,RTMDet-tiny在1020 FPS下仅用4.8M参数就实现了41.1%的AP。RTMDet-s的AP为44.6%,每秒819帧,超过了以前最先进的小型模型。当扩展到实例分割时,RTMDet-Ins-X在基准测试中都获得了实时场景的最新性能,在COCO val集上的掩模AP在180 FPS下分别为44.6%。
表 11. RTMDet-Ins 与其他实例分割算法的性能对比3.2 旋转目标检测旋转目标检测是遥感领域使用最广泛的...
base= 'mmdet::/RTMDet_M_group_fisher/RTMDet_ins_M.py' pretrained_path = '/root/xxxx/best_coco_bbox_mAP_epoch_580.pth' # noqa interval = 10 normalization_type = 'act' lr_ratio = 0.1 target_flop_ratio = 0.5 input_shape = (1, 6, 960, 960) ...