rotation-yolov5:基于yolov5的旋转检测

旋转检测 要求 torch==1.6 shapely==1.7.1 opencv==4.2.0.34 推理 您可以下载（密码为4ud5）以通过我的数据集（而非DOTA）检测船舶，以测试演示。 $ python detect.py 火车 我的模型需要什么格式 与yolo数据集没有太大区别，只是增加一个角度，我们定义box属性w总是长于h！ 因此，我们将框标签定义为（cls，c_x，c_y，最长边，短边，角度） 注意！我们定义角度是一个分类问题，因此我们为角度定义了180个类别。 例如： 角度范围是[-90，90），因此wo在制作数据集标签时应将角度增加90 ，然后标签的范围应为[0,179） 修改yaml models / yolov5m.yaml：将nc设置为您的数据集类num； data / wheat0.yaml：将nc设置为数据集类编号，并将名称设置为数据集类名称； $