PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block
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 s...
Main Authors: | , |
---|---|
Format: | Article |
Language: | English |
Published: |
MDPI AG
2023-08-01
|
Series: | Applied Sciences |
Subjects: | |
Online Access: | https://www.mdpi.com/2076-3417/13/17/9875 |
_version_ | 1827728227365289984 |
---|---|
author | Xiaohu Zhang Haifeng Huang |
author_facet | Xiaohu Zhang Haifeng Huang |
author_sort | Xiaohu Zhang |
collection | DOAJ |
description | 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. |
first_indexed | 2024-03-10T23:27:21Z |
format | Article |
id | doaj.art-11733f6ccfc546d8ae71ecc8f1607ec4 |
institution | Directory Open Access Journal |
issn | 2076-3417 |
language | English |
last_indexed | 2024-03-10T23:27:21Z |
publishDate | 2023-08-01 |
publisher | MDPI AG |
record_format | Article |
series | Applied Sciences |
spelling | doaj.art-11733f6ccfc546d8ae71ecc8f1607ec42023-11-19T07:52:54ZengMDPI AGApplied Sciences2076-34172023-08-011317987510.3390/app13179875PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention BlockXiaohu Zhang0Haifeng Huang1School of Electronics and Communication Engineering, Sun Yat-sen University, Guangzhou 510275, ChinaSchool of Electronics and Communication Engineering, Sun Yat-sen University, Guangzhou 510275, ChinaCrack 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.https://www.mdpi.com/2076-3417/13/17/9875convolutional neural networkimage segmentationcrack detectionU-Netcrack segmentation |
spellingShingle | Xiaohu Zhang Haifeng Huang PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block Applied Sciences convolutional neural network image segmentation crack detection U-Net crack segmentation |
title | PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block |
title_full | PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block |
title_fullStr | PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block |
title_full_unstemmed | PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block |
title_short | PSNet: Parallel-Convolution-Based U-Net for Crack Detection with Self-Gated Attention Block |
title_sort | psnet parallel convolution based u net for crack detection with self gated attention block |
topic | convolutional neural network image segmentation crack detection U-Net crack segmentation |
url | https://www.mdpi.com/2076-3417/13/17/9875 |
work_keys_str_mv | AT xiaohuzhang psnetparallelconvolutionbasedunetforcrackdetectionwithselfgatedattentionblock AT haifenghuang psnetparallelconvolutionbasedunetforcrackdetectionwithselfgatedattentionblock |