关键词:
变压器目标检测
YOLOv8
深度学习
街景图像
摘要:
街景图像是一种城市街道级别信息地理大数据,利用街景图像不仅能够实现大范围、高效率的变压器巡检,还能降低巡检成本。但是,街景图像中的变压器往往像素少、分辨率低、背景复杂,导致现有目标检测方法对变压器的检测准确度不理想。针对上述问题,本文提出一种改进的YOLOv8算法YOLOv8-WSX。首先,使用WIoU作为损失函数,强化模型对困难样本的检测性能;然后,引入空间分组增强(SGE)注意力机制模块,提高模型的特征提取能力;最后,增加微小目标检测头,解决微小变压器目标漏检的问题。实验结果表明,相较于YOLOv8,YOLOv8-WSX的F1值提升了5.9个百分点,IoU阈值为50%时的平均精确率均值提升了6.3个百分点,IoU阈值在50%~95%范围内的平均精确率均值提升了3.2个百分点,且模型的参数量有所下降。