针对现有输电线路绝缘子缺陷检测模型在雨天复杂场景下识别效果差、推理速度慢等问题,在YOLOv7-tiny(you only look once version 7-tiny)的基础上提出了RID-YOLOv7(rain insulator detection-YOLOv7)轻量级雨天场景绝缘子缺陷检测模型...针对现有输电线路绝缘子缺陷检测模型在雨天复杂场景下识别效果差、推理速度慢等问题,在YOLOv7-tiny(you only look once version 7-tiny)的基础上提出了RID-YOLOv7(rain insulator detection-YOLOv7)轻量级雨天场景绝缘子缺陷检测模型。首先探索了坐标注意力机制(coordinate attention,CA)在主干特征提取网络中的最优嵌入位置,提升了模型对目标位置关键特征的提取能力;然后在颈部特征融合网络中引入了幻影混洗卷积(ghost shuffle convolution,GSConv)和幻影混洗跨级部分矢量化旋涡(vortex of vectorized ghost shuffle cross stage partial,VoV-GSCSP),大幅降低了推理时间;最后使用了明智交并比(wise intersection over union,WIoU)优化边界框定位损失函数,提高了模型收敛效率。结果表明,与原始YOLOv7-tiny相比,RID-YOLOv7模型的精确率、召回率和平均精确率均值分别提升了2.41%、5.44%和3.22%,推理速度为88.7帧/s,有效平衡了检测速度和精度。该模型适合对雨天场景下输电线路绝缘子缺陷进行实时检测。展开更多
文摘针对现有输电线路绝缘子缺陷检测模型在雨天复杂场景下识别效果差、推理速度慢等问题,在YOLOv7-tiny(you only look once version 7-tiny)的基础上提出了RID-YOLOv7(rain insulator detection-YOLOv7)轻量级雨天场景绝缘子缺陷检测模型。首先探索了坐标注意力机制(coordinate attention,CA)在主干特征提取网络中的最优嵌入位置,提升了模型对目标位置关键特征的提取能力;然后在颈部特征融合网络中引入了幻影混洗卷积(ghost shuffle convolution,GSConv)和幻影混洗跨级部分矢量化旋涡(vortex of vectorized ghost shuffle cross stage partial,VoV-GSCSP),大幅降低了推理时间;最后使用了明智交并比(wise intersection over union,WIoU)优化边界框定位损失函数,提高了模型收敛效率。结果表明,与原始YOLOv7-tiny相比,RID-YOLOv7模型的精确率、召回率和平均精确率均值分别提升了2.41%、5.44%和3.22%,推理速度为88.7帧/s,有效平衡了检测速度和精度。该模型适合对雨天场景下输电线路绝缘子缺陷进行实时检测。