基于旋转框标注YOLOV3的SAR舰船目标检测方法
摘要:
本发明提供一种基于旋转框标注YOLOV3的SAR舰船目标检测方法,包括如下步骤:S1、采集包含感兴趣目标的SAR图像,对所述SAR图像进行预处理并以(x1,y1,x2,y2,x3,y3,x4,y4)8参数描述任意角度的目标框;S2、将所述8参数目标框转化为5参数目标框,即为(trx,try,trw,trh,trθ);S3、改进YOLOv3模型:YOLOv3模型中的head部分在输出时将各尺度特征图检测分支的输出维数增加到[S,S,B*(c+10)]。本发明针对现有技术中目标检测算法存在的不足,如在复杂背景下对遥感图像中的飞机目标检测处理步骤过多,实时性较差的问题,而在yolov3网络中加入旋转框并没有增加过多的工作量,且通过旋转,倾斜目标的准确率也得到了一定的提升。
0/0