Yolov5 旋转检测 - 源码。

简介：
船舶旋转检测需要使用torch版本1.6和shapely版本1.7.1，同时依赖opencv版本4.2.0.34进行推理。 您可以通过提供的密码“4ud5”下载相关资源，以便利用我的数据集（而非DOTA）进行船舶检测，并用于演示测试。 我的模型对输入数据的格式要求与YOLO数据集较为相似，主要区别在于增加了角度信息。 具体而言，我们通过定义box属性w始终大于h来规范框的标签。 此外，我们将框标签定义为（cls, c_x, c_y, 最长边, 短边, 角度），其中角度被视为一个分类问题，因此我们为角度定义了180个类别。 角度范围限定在[-90, 90]，因此在数据集标注过程中，需要将角度增加90度，并调整标签范围为[0, 179]。 请修改yaml文件 models / yolov5m.yaml：设置nc参数为您的数据集类别的数量； 同时修改 data / wheat0.yaml：设置nc参数为数据集类别的编号，并将其名称设置为相应的类名。