一种基于卷积神经网络的遥感图像舰船目标检测方法

    公开(公告)号:CN116563726A

    公开(公告)日:2023-08-08

    申请号:CN202310520115.2

    申请日:2023-05-08

    Abstract: 本发明提供一种基于卷积神经网络的遥感图像舰船目标检测方法,包括构建遥感舰船图像数据集,对舰船目标进行标注;对数据集进行数据增强,划分训练集、验证集和测试集;构建基于改进RetinaNet的目标检测网络模型;初始化网络,设置训练参数,使用训练集训练目标检测网络模型;将测试集图像输入训练后的目标检测网络模型,预测舰船目标的分类和回归结果。本发明基于改进RetinaNet进行舰船目标检测;通过在特征金字塔中嵌入坐标注意力模块,强化多尺度特征融合过程,增加网络对复杂目标的关注度;通过构建方向不变模型,生成深度旋转不变特征,提高网络对旋转目标的适应能力;通过定义基于KLD的回归损失函数,解决边界不连续性问题,提升网络对密集目标的检测能力。

Patent Agency Ranking