Abstract
The automatic detection of the degree of surface corrosion on metal structures is of significant importance for assessing structural damage and safety. To effectively identify the corrosion status on the surface of coastal metal facilities, this study proposed a CBG-YOLOv5s model for metal surface corrosion detection, based on the YOLOv5s model. Firstly, we integrated the Convolutional Block Attention Module (CBAM) into the C3 module and developed the C3CBAM module. This module effectively enhanced the channel and spatial attention capabilities of the feature map, thereby improving the feature representation. Second, we introduced a multi-scale feature fusion concept in the feature fusion part of the model and added a small target detection layer to improve small target detection. Finally, we designed a lighter C3Ghost module, which reduced the number of parameters and the computational load of the model, thereby improving the running speed of the model. In addition, to verify the effectiveness of our method, we constructed a dataset containing 6000 typical images of metal surface corrosion and conducted extensive experiments on this dataset. The results showed that compared to the YOLOv5s model and several other commonly used object detection models, our method achieved superior performance in terms of detection accuracy and speed.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.