基于SAR/AIS跨模态融合YOLO网络的弱小舰船检测方法

本发明公开了一种基于SAR/AIS跨模态融合YOLO网络的弱小舰船检测方法，包括：1.获取SAR图像，通过与之匹配的全年AIS数据计算出全年舰船分布密度，制作小舰船检测样本集；2.在数据输入阶段设计跨模态差异性融合模块，为模型提供复杂环境下的跨模态舰船先验信息；3.在特征提取阶段构建特征增强模块用于增强对小舰船特征的特征表示，在特征恢复阶段构建通道重建模块，增强了特征交互学习，从而使模型能更好地聚焦于关键特征，4.将数据集输入网络中进行训练；5.将待分类的SAR图像输入训练好的模型，得到舰船检测结果。本发明能有效解决在港口、河道等复杂环境下的小型舰船检测效果不佳问题，且计算效率高，具有较好的工程应用价值。