0
点赞
收藏
分享

微信扫一扫

[深度学习][原创]旋转目标检测框架yolov5_obb,paddledetection-s2anet和mmrotate谁最好用?


测试时间:2022年3月10日

测试框架:yolov5_obb,paddledetection-s2anet和mmrotate

个人觉得安装难度(从难到易)mmrotate>yolov5_obb>paddledetection

测试对象:roLabelImg标注的数据集100张条码图片

测试地址:

yolov5_obb:GitHub - hukaixuan19970627/yolov5_obb: yolov5 + csl_label.(Oriented Object Detection)(Rotation Detection)(Rotated BBox)基于yolov5的旋转目标检测

paddledetection:GitHub - PaddlePaddle/PaddleDetection: Object Detection toolkit based on PaddlePaddle. It supports object detection, instance segmentation, multiple object tracking and real-time multi-person keypoint detection.

mmrotate:https://github.com/open-mmlab/mmrotate

测试yolov5_obb框架感受:

测试paddledetection-s2anet感受:

我使用的是paddledetection2.3版本,数据集需要自己转换coco格式,测试效果不错,收敛较快,检测结果还可以。总之好评

测试mmrotate感受:

测试mmrotate为0.1.0版本,如果用mim安装很快,但是国内问题,如果用源码就安装很困难。我使用的是上s2anet这个配置文件,数据集需要自己roLableImg转dota格式,测试效果不错,收敛很快,训练4-5 epoch,ap显示为1。检测结果很棒,和ap对应一个样。此外mmrotate出来不长,有些小BUG需要改改代码。其中有个报version错误,看来是mmdetection和mmrotate版本不兼容导致。稍微改改就行。

以上算是目前旋转框常用框架。个人不推荐yolov5_obb感觉效果很差。按照好用程度我排名次(从好用到不好用)

mmrotate>paddledetection>yolov5_obb

举报

相关推荐

0 条评论