PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block

Abstract

Crack detection is an important task for road maintenance. Currently, convolutional neural-network-based segmentation models with attention blocks have achieved promising results, for the reason that these models can avoid the interference of lights and shadows. However, by carefully examining the structure of these models, we found that these segmentation models usually use down-sampling operations to extract high-level features. This operation reduces the resolution of features and causes feature information loss. Thus, in our proposed method, a Parallel Convolution Module (PCM) was designed to avoid feature information loss caused by down-sampling. In addition, the attention blocks in these models only focused on selecting channel features or spatial features, without controlling feature information flow. To solve the problem, a Self-Gated Attention Block (SGAB) was used to control the feature information flow in the attention block. Therefore, based on the ideas above, a PSNet with a PCM and SGAB was proposed by us. Additionally, as there were few public datasets for detailed evaluation of our method, we collected a large dataset by ourselves, which we named the OAD_CRACK dataset. Compared with the state-of-the-art crack detection method, our proposed PSNet demonstrated competitive segmentation performance. The experimental results showed that our PSNet could achieve accuracies of 92.6%, 81.2%, 98.5%, and 76.2% against the Cracktree200, CRACK500, CFD, and OAD_CRACK datasets, respectively, which were 2.6%, 4.2%, 1.2%, and 3.3% higher than those of the traditional attention models.