Rotation-YOLOv5 是一个基于 YOLOv5 的改进版本项目,专注于提高对旋转物体的检测精度。该项目提供了完整的源代码以便于研究和开发。
旋转检测需要使用torch==1.6, shapely==1.7.1 和 opencv==4.2.0.34进行推理。您可以下载相关文件(密码为4ud5)并通过我的数据集来测试船舶的检测效果,而非DOTA。
您的模型需采用特定格式的数据集,与Yolo数据集类似,但增加了角度信息。我们定义框属性w总是长于h,并将框标签定义为(cls, c_x, c_y, 最长边, 短边, 角度)的形式。注意:我们将角度视为分类问题,在制作数据集时需将角度值加上90(即范围是[0,179)),以符合我们的定义。
修改yaml文件:
- 在models/yolov5m.yaml中设置nc为您的数据集中类的数量;
- 在data/wheat0.yaml中同样设置nc为您数据集中的类别数量,并且将名称项设为相应类别的名字。