LIAO Xiaohui, XIONG Zongyi, KONG Bin, XIE Zichen, LIU Xiangyang, GAO Ziyang
Abstract:
In order to improve the detection method of heating defects of electrical equipment and the recognition accuracy of the algorithm for infrared images of electrical equipment, a method of infrared image detection of electrical equipment based on data enhancement and improved YOLOv5 was proposed. Firstly, for the problem of low signal-to-noise ratio and low contrast of infrared images of electrical equipment, a fast guided filtering algorithm was used to denoise the infrared images in the data set. The CLAHE algorithm was improved by introducing Gamma correction, and then the contrast of infrared images was enhanced. The premise of detecting the heating defects of electrical equipment was to accurately identify and classify the equipment. Then, in order to improve the accuracy of the detection algorithm, the information aggregation and distribution mechanism was introduced to improve the feature fusion module based on the original YOLOv5 algorithm, which enhanced the multi-scale feature fusion ability. Meanwhile, the Focal-CIoU loss function was also introduced to make the algorithm pay more attention to high-quality samples and suppress low-quality samples, which enhanced the rate of convergence for the model. It was verified that the mAP value of the improved algorithm on the self-built data set was 93.6%, which was 4.0 percentage points higher than that before the improvement. Compared with Faster R-CNN, SSD, YOLOv3 and YOLOv7, the mAP value of the proposed algorithm was increased by 4.5 percentage points, 6.1 percentage points, 4.7 percentage points and 3.5 percentage points respectively, and the frame rate reached 32 frames per second, which could meet the real-time recognition requirements of electrical equipment.