郑州大学学报（工学版）2024，Vol.45Issue(4)：117-124,8.DOI:10.13705/j.issn.1671-6833.2024.04.010
基于轻量化YOLOv5的电气设备外部缺陷检测
Electrical Equipment External Defect Detection Based on Lightweight YOLOv5
摘要
Abstract
In order to improve the accuracy of real-time detection of external defects of electrical equipment in sub-stations and make the detection model more lightweight,a lightweight YOLOv5 based external defect detection method for electrical equipment was proposed.Firstly,the external defect image dataset of electrical equipment was constructed and processed by data enhancement.Secondly,three optimization strategies were used to improve the original YOLOv5.The EfficientViT network was introduced to improve the backbone network of the algorithm to re-duce the number of model parameters,and the SimAM parameter-free attention mechanism was added to the Neck part of the algorithm to improve the recognition accuracy with the complex background of the substation.At the same time,the Soft-NMS module was used to improve the screening method of the detection box to avoid the phe-nomenon of defect missed detection.Finally,verified by ablation test,the mAP value of the lightweight external de-fect detection model of electrical equipment was stable at 86.4%,which was 1.2 percentage points higher than that of the original model,the number of model parameters were reduced by 20%,the calculation amount was reduced by 38%,and the model size was 11 MB,which was 19.7%lower than that of the original model.The improved model could meet the requirements of real-time detection of external defects of equipment.关键词
缺陷检测/电气设备/轻量化YOLOv5/EfficientViT网络/SimAM注意力/Soft-NMS结构Key words
defect detection/electrical equipment/lightweight YOLOv5/EfficientViT network/SimAM attention/Soft-NMS structure分类
信息技术与安全科学引用本文复制引用
廖晓辉,谢子晨,辛忠良,陈怡,叶梁劲..基于轻量化YOLOv5的电气设备外部缺陷检测[J].郑州大学学报（工学版）,2024,45(4):117-124,8.基金项目
国家自然科学基金资助项目(52307227) （52307227）
河南省自然科学基金资助项目(232300421198) （232300421198）
河南省科技攻关项目(222102220053) （222102220053）
