search

ming71 / rotate-yolov3

Arbitrary oriented object detection implemented with yolov3 (attached with some tricks).
222 stars 58 forks source link

非常感谢您的 CUDA RNMS，已经跑起来了！请问有没有比较好的 CUDA NMS 代码可以参考？ #11

Closed clw5180 closed 4 years ago

clw5180 commented 4 years ago

RT

jingweirobot commented 4 years ago

@clw5180 in the original repo-YOLOv3, these guys provided several nms methods, I have tried already, but it is not good enough. Maybe you go ahead to test them.

tutu96177 commented 4 years ago

有 pipline 可以共享一下吗？跑起来的步骤。

clw5180 commented 4 years ago

有 pipline 可以共享一下吗？跑起来的步骤。

RNMS我单独拿出来用了，完整的项目我没跑

jingweirobot commented 4 years ago

@clw5180 有编译RNMS的pipiline吗？可以分享一下吗，我现在用的系统没有调过去。thanks in advance

ming71 commented 4 years ago

RT

CUDA NMS很多地方都有不难找，你可以试试mmdetection的，写的比较新比较好读，调用也简单。

ming71 commented 4 years ago

有 pipline 可以共享一下吗？跑起来的步骤。

返校后有时间我会捋一下程序文档重新commit，半年没跑这个codebase我也有点忘了。

jingweirobot commented 4 years ago

有 pipline 可以共享一下吗？跑起来的步骤。

返校后有时间我会捋一下程序文档重新commit，半年没跑这个codebase我也有点忘了。

哈哈，你不用缕，这两天我们花时间可以屡一下，缕的满意，发给你，你upload

ming71 commented 4 years ago

有 pipline 可以共享一下吗？跑起来的步骤。

返校后有时间我会捋一下程序文档重新commit，半年没跑这个codebase我也有点忘了。

哈哈，你不用缕，这两天我们花时间可以屡一下，缕的满意，发给你，你upload

老哥如果实在不行可以试试cython加速的overlap计算：dota_devkit，可以计算多边形的overlap，然后自己除一下union就行了。 这个速度不逊色于cuda，效果也比我的这个rnms稍微准确一点，而且用的人很多应该不会出奇怪的bug。

原程序还在ORN上修改了一下支持半精度训练和更新版的pytorch之类的，回头开学了update上。

clw5180 commented 4 years ago

@clw5180 有编译RNMS的pipiline吗？可以分享一下吗，我现在用的系统没有调过去。thanks in advance

这个老哥有个CUDA的项目里面讲了