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

Posted 2022-03-11 FL1623863129

测试时间：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框架感受：

作者在2022年1月更新，测试训练算简单，但是收敛很慢，我训练集90张测试集10张，训练epoch到30才有很低map，前面10个epoch几乎P,R,map全是0，测试发现检测效果很差，出现漏检，多框现象严重。看了github issue发现类别单类或者多类训练区别很大，我训练单类theta值很大后面才会慢慢降下去，但是设置>2类，theta很小，但是效果很差。测试结果不尽如意。

测试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