旋转检测
要求
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设置为数据集类编号,并将名称设置为数据集类名称;
$
1