关键词:
小目标检测
YOLOv8
输电线路
特征融合
边界框回归损失函数
摘要:
输电线路易受自然环境影响导致部件老化或者损坏问题,因此对输电线路的巡检至关重要。利用无人机航拍图像结合深度学习进行图像识别已成为输电线路巡检的主流方法。然而,输电线路本身包含许多小目标部件,加上拍摄距离的不确定性,导致航拍图像出现大量小目标。由于小目标可用信息不足且定位难度大,造成输电线路小目标检测精度低。针对以上问题,本文在YOLOv8算法的基础上进行了一系列改进。
一方面,针对输电线路小目标像素点少导致目标信息不足的问题,在YOLOv8的基础上,提出了基于特征改进的网络结构:首先,提出了一种新型的注意力机制ECBAM,并将其嵌入到YOLOv8的C2f模块中,形成了C2f-ECBAM模块,以进一步增强网络对特征的提取能力;其次,设计了一个跨层特征图信息交互模块CVT,用以替换骨干网络和颈部网络FPN横向连接的Concat模块,使得深层特征图与浅层特征图之间建立信息交互能力,充分利用浅层特征图包含的小目标信息;最后,在YOLOv8原有的三个检测头基础上,增加了一个可输出更高维度特征图的检测头SOD-H,以更好地适应小目标检测。将融合以上改进方法的YOLOv8命名为YOLOv8s-SODT。
另一方面,针对输电线路小目标定位难度大的问题,在YOLOv8s-SODT的基础上,提出了一个全新的边界框回归损失函数CFIo U Loss,进一步提升YOLOv8s-SODT的定位能力:首先,CFIo U Loss通过在Io U Loss基础上加入角点距离偏差的惩罚项,增强了位置参数的更新能力,有效克服了传统Io U系列损失函数在预测框与目标框中心接近或重叠时退化为Io U Loss的问题;其次,CFIo U Loss还加入了目标的空间信息,即前景区域,以区分不同的回归效果,使得定位不准确的预测框能够更精确地适配目标框。将融合了CFIo U Loss的YOLOv8s-SODT算法命名为YOLOv8s-SODT+。
通过多组对比实验表明,相较于YOLOv8s基线模型,YOLOv8s-SODT算法在自建的输电线路小目标测试集上的所有指标均得到提升,特别在小目标平均精确率均值(APSmall)上取得了最佳提升效果。融合了全新的边界框回归损失函数CFIo U Loss的YOLOv8s-SODT,即YOLOv8s-SODT+,在所有指标上均也得到了提升。此外,YOLOv8s-SODT和CFIo U Loss在公共小目标数据集Vis Drone2019上也展现出了卓越的性能,与其他模型相比具有明显优势,这进一步验证了YOLOv8s-SODT和CFIo U Loss良好的泛化性能。结果表明,本文提出的输电线路小目标检测算法具有一定的学术创新性和工程应用价值。