一种改进的SSD红外舰船目标检测算法

作者:王岩; 娄树理*
来源:烟台大学学报(自然科学与工程版), 2023, 36(04): 487-493.
DOI:10.13951/j.cnki.37-1213/n.220421

摘要

针对海战场舰船目标检测精度和速度要求高的技术难题,提出一种改进的SSD红外舰船目标检测算法,通过改进ResNet-50网络以提高整体网络性能;引入Mosaic数据增强方法对图像进行数据增强,提高检测效率和丰富检测物体的背景;引入ECANet通道注意力机制,提高对舰船目标的识别能力,降低舰船目标的漏检率和误检率;使用余弦退火衰减学习率来优化网络。实验结果表明,在保证检测速度的基础上,改进后算法的检测精度均值达到98.8%,对红外舰船目标有着很好的检测效果。