基于PCBAM-YOLOv5网络的变电站表计缺陷检测

作者：高志国; 袁牧; 童旸; 倪修峰; 程汪刘

来源：电子设计工程, 2023, 31(04): 77-81.

DOI：10.14022/j.issn1674-6236.2023.04.016

• 表计
• YOLOv5
• 缺陷检测
• 注意力机制

准确检测变电站中表计缺陷并及时更换是保证电力系统安全运行的重要措施。文中针对表计缺陷图像背景复杂、目标尺寸不一、表计部分遮挡等问题，设计了一种端到端的PCBAM-YOLOv5网络用于表计缺陷检测。为了使网络模型关注复杂背景下表计关键区域，在YOLOv5的基础上，加入并联型混合注意力模块，将通道注意力模块、空间注意力模块和输入特征图并联，在聚焦表计特征的同时突出表计缺陷特征。实验结果表明，PCBAM-YOLOv5在电表缺陷检测测试集上的mAP可达到78%，相对于YOLOv5提高了1.6%。

• 单位
  电子信息工程学院; 安徽南瑞继远电网技术有限公司; 国网安徽省电力有限公司; 安徽大学