科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明涉及一种SAR舰船旋转目标检测方法,包括：获取待检测SAR图像舰船数据；将待检测SAR图像舰船数据输入训练好的旋转目标检测模型进行推理,得到舰船目标的角度、位置及类别信息；旋转目标检测模型基于卷积块注意力机制的YOLOv5网络和旋转矩形框的中点偏移量表示法构建得到；训练好的旋转目标检测模型为基于预先构建的训练集、并采用适应于旋转检测的多任务联合网络损失函数对旋转目标检测模型进行训练得到；预先构建的训练集基于原始SAR图像舰船样本数据和标注有待检测船舰目标所在旋转矩形框的标签文件构建得到。该方法能避免损失边界不连续性,具备轻量级网络参数量少的优点,可以实现SAR图像舰船目标的任意方向检测,并获得舰船相应的角度信息。
- 单位
