实海域实时小尺度目标检测算法 |
| |
作者姓名: | 冯辉 蒋成鑫 丁轶航 徐海祥 |
| |
作者单位: | 1. 武汉理工大学高性能船舶技术教育部重点实验室;2. 武汉理工大学船海与能源动力工程学院 |
| |
基金项目: | 国家自然科学基金资助项目(51879210,51979210);;中央高校基本科研业务费专项资金资助项目(2019Ⅲ040,2019Ⅲ132CG); |
| |
摘 要: | 针对实海域可见光图像中海天线附近的远距离小目标特征少、检测精度低的问题,设计了一种基于YOLOv5的实海域实时目标检测算法.首先,通过在网络结构中设置双向特征融合结构,结合高层特征图的强语义信息和低层特征图的定位信息,提高了小目标的检测精度;其次,引入双通道注意力机制,并改进权重分配函数,提高了小目标的检测召回率;然后,通过新增极小目标检测层,增加对小目标物体的检测精度;最后,通过实海域数据集对算法进行验证,证明了算法对小尺度目标检测精度提升明显,同时保持了良好的实时性.
|
关 键 词: | 可见光图像 小尺度目标 目标检测 双向特征融合 双通道注意力机制 |
|
|