Insulator defect detection algorithm based on attention mechanism
With the rapid development of deep learning, using object detection algorithms to detect defects in aerial insulator images has become the main way of insulator inspection. In response to the problems of low detection accuracy of small targets, weak representation ability of feature maps, and limited extraction of key information in traditional object detection algorithms, this paper introduces the ECA attention module into the YOLOv51 model, which can effectively enhance and suppress a large amount of feature information at the channel level of the feature map, improve the network's attention to defect areas, and fuse shallow and deep information of insulators to avoid the loss of feature information and reduce the incidence of missed detections. The improved model has improved the accuracy of insulator defect detection, especially in identifying small target defects.
Authors
- Xiaomian Li (ORCID: https://orcid.org/0000-0002-0938-9329)
- Eric Blancaflor (ORCID: https://orcid.org/0000-0002-7189-3040)
Institutions
- Mapúa University (PH)
Publication Details
- Published
- 2024-11-08
- DOI
- https://doi.org/10.1117/12.3049591
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00