一种基于改进YOLOv4算法的遥感图像海上船舶识别系统及其方法

摘要

本发明公开了一种基于改进YOLOv4算法的遥感图像海上船舶识别系统及其方法，包括：搜集以往拍摄的或搜集的海面场景的卫星遥感图像；使用数据标记软件将预处理后的图片进行种类标注；将遥感图片中的船舰与周边环境物相分割，以消除图像噪声；获得YOLO算法锚点框的估计值；生成YOLOv4框架；生成YOLOv4的检测框；设定候选框的阈值，最终得到预测框；计算三种损失函数并使其总值最小，得到训练后改进的YOLOv4神经网络；将测试子集中图片输入到训练后改进的YOLOv4网络，获得目标类别、目标在图片中具体地位置和目标宽高，完成目标检测。本发明对海面舰船目标可快速检测及自动识别，舰船识别概率和准确率高。