重庆理工大学学报(自然科学)

• •    

一种SSD-RTF输电线路小目标检测算法研究

  

  1. 河北科技大学 信息科学与工程学院,石家庄 050018

摘要: 针对传统输电线路无人机巡检图像检测方法对输电线路小目标检测能力弱,并且存在错检和漏检率高、浅层网络语义信息不足等问题,提出了一种SSD-RTF输电线路小目标检测算法模型。在原始的SSD算法主干网络VGG-16浅层网络层添加视觉机制来扩大感受野,并且引入三叉乾特征融合模块提取特征图的多尺度特征,增加特征图的鲁棒性;融合FusionNet浅层特征模块,从而增加小目标的提取能力;使用注意力机制以提升对关键信息的学习效果,以此来提高目标检测效率;改进非极大值抑制提高网络的表示能力。改进后的SSD-RTF算法在自行构建的输电线数据集上的实验结果显示,在检测小目标时准确性和实时性方面都有了一定的提高,整体mAP上升了7.5%,同时也减少了错检和漏检情况。

关键词: 目标检测, 浅层融合, 三叉乾特征融合, 视觉机制, 非极大值抑制