摘要
针对YOLOv3目标检测算法在遥感图像小目标检测中,存在检测准确度低和误检率高的问题,提出了一种基于改进YOLOv3的小目标检测算法:MS-YOLOv3.为了提高小目标特征提取的能力,在原YOLOv3的3个检测尺度基础上,增加了104×104的尺度,有效地掌握了样本的特点;同时在特征提取阶段,把通道注意力模块(MS-CAM)加入原网络中的残差块内,MS-CAM通过两个不同尺度的分支来获得通道注意力,使得语义和尺度不一致的特征能更好地融合,从而更准确地检测出小目标.实验表结果明,MS-YOLOv3算法能够有效抑制复杂背景的干扰,目标区域位置更加准确,检测的精确度更高、误检率更低,更适合遥感图像小目标的检测.
- 单位