基于改进的YOLOv7的雾天场景下绝缘子及其自爆缺陷检测方法
DOI:
CSTR:
作者:
作者单位:

三峡大学电气与新能源学院

作者简介:

通讯作者:

中图分类号:

TM216

基金项目:

国家自然科学基金项目(52107108)


Detection method of insulator and self-explosion defect in foggy scene based on improved YOLOv7
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了解决现有的目标检测方法在雾天场景下存在绝缘子、自爆缺陷识别准确率低、易漏检的问题,提出了一种融合了坐标注意力机制(CA)和双向加权特征金字塔(BiFPN)的YOLOv7的雾天场景下绝缘子、自爆缺陷检测方法。首先,通过大气散射模型采用中心点合成雾的方法生成轻雾数据集、浓雾数据集和混合雾气浓度数据集;其次,在主干特征提取网络末端和预测端前端融入坐标注意力机制,提高网络对重要特征的关注程度;再次,在特征融合网络中借鉴BiFPN的思想添加跨层权重连接,提升模型的特征融合能力,改善遮挡目标与小目标的漏检问题;最后,考虑到真实框与预测框之间的方向匹配问题,使用SIoU损失函数替代CIoU损失函数,进一步提升模型的检测性能。研究结果表明:改进后的YOLOv7在轻雾、浓雾和混合雾气状态下的平均精确率分别达到96.95%、95.58%和96.65%,与原始YOLOv7相比分别提升了6.65%、5.55%和6.54%,证明了改进后的YOLOv7在雾天环境下绝缘子、自爆缺陷有较好的的检测效果。

    Abstract:

    In order to solve the problems of low recognition accuracy and easy missed detection of insulators and self-explosion defects in the existing target detection methods in foggy scenes, a YOLOv7 insulator and self-explosion defect detection method in foggy scenes is proposed, which combines coordinate attention mechanism (CA) and bidirectional weighted feature pyramid (BiFPN). Firstly, the light fog data set, the dense fog data set and the mixed fog concentration data set are generated by using the center point synthesis fog method through the atmospheric scattering model. Secondly, the coordinate attention mechanism is integrated into the end of the backbone feature extraction network and the front end of the prediction end to improve the attention of the network to important features. Thirdly, in the feature fusion network, the idea of BiFPN is used to add cross-layer weight connection to improve the feature fusion ability of the model and improve the missed detection of occluded targets and small targets. Finally, considering the direction matching problem between the real box and the prediction box, the SIoU loss function is used to replace the CIoU loss function to further improve the detection performance of the model. The results show that the average accuracy of the improved YOLOv7 in light fog, dense fog and mixed fog is 96.95 %, 95.58 % and 96.65 %, respectively, which is 6.65 %, 5.55 % and 6.54 % higher than that of the original YOLOv7. It is proved that the improved YOLOv7 has better detection performance for insulators and self-explosion defects in foggy environment.

    参考文献
    相似文献
    引证文献
引用本文
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-06-02
  • 最后修改日期:2023-07-31
  • 录用日期:2023-08-08
  • 在线发布日期:
  • 出版日期:
文章二维码