计算机工程与应用››2024,第60卷››发行(10): 256-265.内政部:10.3778/j.issn.1002-8331.2309-0468

图形图像处理 • 上一篇   下一篇

电力巡检中改进YOLOv5系列的缺陷检测算法研究

王磊,郝涌汀,潘明然,赵慕东,张永鑫,张茗宇  

  1. 1沈阳理工大学 机械工程学院,沈阳 110159
    2.辽沈工业集团有限公司 研发中心,沈阳 110045
  • 出版日期:2024-05-15 发布日期:2024-05-15

基于YOLOv5s改进的电力检测缺陷检测算法

王磊、郝永亭、潘明然、赵慕东、张永新、张明宇  

  1. 1.沈阳理工大学机械工程学院,沈阳110159
    2.中国沈阳辽沈实业集团有限公司研发中心,110045
  • 在线:2024-05-15 出版:2024-05-15

摘要:针对无人机进行电力巡检时关键零件的检测精度较低的问题,提出了一种基于YOLOv5系列的改进型缺陷检测算法。在骨干网络中引入卷积神经网络注意力模块(CBAM)增强网络对特征图中重要信息的提取效率;将YOLOv5s中国PANet特征融合框架替换为双向特征金字塔网络(BiFPN)引入可学习的权重,映射不同的学习特征,增加对贡献较大特征的映射。在空间金字塔池化模块(空间金字塔池,SPP)的基础上加入上下文卷积模块,提升特征的表达能力。通过构建航拍数据集进行实验验证,结果表明,改进后的算法mAP 95.6%、93.7%、93.8%为进一步验证算法在嵌入式系统的运行效果,通过缩小网络宽度进行轻量化,利用张力RT推理引擎,优化了网络结构并加速了模型的推理。将模型加速后部署至Jetson Xavier NX公司平台进行测试,单帧图像平均运行时间为24.6?ms、90.8%、90.5%、Jetson Xavier NX设备上对目标实现精准识别。改进后的模型提高了检测精度,体现了算法的有效性,满足电力巡检作业的实时检测需求。

关键词: 电力巡检, 目标检测, 注意力机制, 特征融合, YOLO公司

摘要:提出了一种基于YOLOv5的改进型缺陷检测算法,以解决无人机电力线检测过程中关键部件检测精度低的问题。该算法在骨干网络中引入了卷积神经网络注意力模块(CBAM),以提高从特征图中提取重要信息的效率。YOLOv5中原有的PANet特征融合框架被双向特征金字塔网络(BiFPN)取代,该网络结合了可学习的权重来映射不同的特征贡献,从而提高了重要特征映射的重要性。此外,在空间金字塔池(SPP)模块的顶部添加了上下文卷积模块,以提高特征表示能力。利用航空摄影数据集进行了实验验证,结果表明,改进后的算法实现了95.6%的mAP,93.7%的准确率,93.8%的召回率。为了进一步验证算法在嵌入式系统上的性能,该模型被加速并部署在Jetson Xavier NX平台上,单帧图像的平均运行时间为24.6?ms的检测准确率为90.8%,召回率为90.5%。此功能允许在Jetson Xavier NX设备上精确识别物体。改进后的模型提高了检测精度,证明了算法的有效性,满足了电力线检测的实时检测要求。

关键词: 电力检查, 目标检测, 注意机制, 特征融合, YOLO公司