Research on Tunnel Crack Detection Algorithm Based on Improved YOLOv5

Jiale Yu,Zhenhai Zhang
DOI: https://doi.org/10.1109/ICCTIT60726.2023.10435797
2023-11-24
Abstract:Aiming at the misjudgment, omission, and insufficient feature extraction ability of the existing deep learning for tunnel crack detection, an improved algorithm model based on YOLOv5 network is proposed. Firstly, the CBAM attention mechanism is introduced to focus on the image successively in channel dimension and spatial dimension, so as to highlight the crack features, and then the BotNet network structure is added into the backbone network of YOLOv5, and the location information between the crack features is interconnected by utilizing the combination of the convolutional neural network and the global self-attention mechanism in the BotNet network. The results show that the improved algorithm proposed in this paper can better extract the crack region information compared with the YOLOv5 model, can recognize cracks of different sizes, and has higher detection accuracy.
Engineering,Materials Science,Computer Science
What problem does this paper attempt to address?