实海域实时小尺度目标检测算法

作者:冯辉; 蒋成鑫; 丁轶航; 徐海祥*
来源:华中科技大学学报(自然科学版)科技大学, 2023, 51(04): 24-29.
DOI:10.13245/j.hust.230404

摘要

针对实海域可见光图像中海天线附近的远距离小目标特征少、检测精度低的问题,设计了一种基于YOLOv5的实海域实时目标检测算法.首先,通过在网络结构中设置双向特征融合结构,结合高层特征图的强语义信息和低层特征图的定位信息,提高了小目标的检测精度;其次,引入双通道注意力机制,并改进权重分配函数,提高了小目标的检测召回率;然后,通过新增极小目标检测层,增加对小目标物体的检测精度;最后,通过实海域数据集对算法进行验证,证明了算法对小尺度目标检测精度提升明显,同时保持了良好的实时性.

全文