Road crack segmentation using an attention residual U-Net with generative adversarial
This paper proposed an end-to-end road crack segmentation model based on attention mechanism and deep FCN with generative adversarial learning. We create a segmentation network by introducing a visual attention mechanism and residual module to a fully convolutional network(FCN) to capture richer loc...
Guardado en:
Autores principales: | , , |
---|---|
Formato: | article |
Lenguaje: | EN |
Publicado: |
AIMS Press
2021
|
Materias: | |
Acceso en línea: | https://doaj.org/article/ccc0569333544ed7a9cc36fc49156950 |
Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
id |
oai:doaj.org-article:ccc0569333544ed7a9cc36fc49156950 |
---|---|
record_format |
dspace |
spelling |
oai:doaj.org-article:ccc0569333544ed7a9cc36fc491569502021-11-29T06:31:36ZRoad crack segmentation using an attention residual U-Net with generative adversarial10.3934/mbe.20214731551-0018https://doaj.org/article/ccc0569333544ed7a9cc36fc491569502021-11-01T00:00:00Zhttps://www.aimspress.com/article/doi/10.3934/mbe.2021473?viewType=HTMLhttps://doaj.org/toc/1551-0018This paper proposed an end-to-end road crack segmentation model based on attention mechanism and deep FCN with generative adversarial learning. We create a segmentation network by introducing a visual attention mechanism and residual module to a fully convolutional network(FCN) to capture richer local features and more global semantic features and get a better segment result. Besides, we use an adversarial network consisting of convolutional layers as a discrimination network. The main contributions of this work are as follows: 1) We introduce a CNN model as a discriminate network to realize adversarial learning to guide the training of the segmentation network, which is trained in a min-max way: the discrimination network is trained by maximizing the loss function, while the segmentation network is trained with the only gradient passed by the discrimination network and aim at minimizing the loss function, and finally an optimal segmentation network is obtained; 2) We add the residual modular and the visual attention mechanism to U-Net, which makes the segmentation results more robust, refined and smooth; 3) Extensive experiments are conducted on three public road crack datasets to evaluate the performance of our proposed model. Qualitative and quantitative comparisons between the proposed method and the state-of-the-art methods show that the proposed method outperforms or is comparable to the state-of-the-art methods in both F1 score and precision. In particular, compared with U-Net, the mIoU of our proposed method is increased about 3%~17% compared with the three public datasets.Xing HuMinghui Yao Dawei ZhangAIMS Pressarticleroad crack segmentationattention residual u-netadversarial learningBiotechnologyTP248.13-248.65MathematicsQA1-939ENMathematical Biosciences and Engineering, Vol 18, Iss 6, Pp 9669-9684 (2021) |
institution |
DOAJ |
collection |
DOAJ |
language |
EN |
topic |
road crack segmentation attention residual u-net adversarial learning Biotechnology TP248.13-248.65 Mathematics QA1-939 |
spellingShingle |
road crack segmentation attention residual u-net adversarial learning Biotechnology TP248.13-248.65 Mathematics QA1-939 Xing Hu Minghui Yao Dawei Zhang Road crack segmentation using an attention residual U-Net with generative adversarial |
description |
This paper proposed an end-to-end road crack segmentation model based on attention mechanism and deep FCN with generative adversarial learning. We create a segmentation network by introducing a visual attention mechanism and residual module to a fully convolutional network(FCN) to capture richer local features and more global semantic features and get a better segment result. Besides, we use an adversarial network consisting of convolutional layers as a discrimination network. The main contributions of this work are as follows: 1) We introduce a CNN model as a discriminate network to realize adversarial learning to guide the training of the segmentation network, which is trained in a min-max way: the discrimination network is trained by maximizing the loss function, while the segmentation network is trained with the only gradient passed by the discrimination network and aim at minimizing the loss function, and finally an optimal segmentation network is obtained; 2) We add the residual modular and the visual attention mechanism to U-Net, which makes the segmentation results more robust, refined and smooth; 3) Extensive experiments are conducted on three public road crack datasets to evaluate the performance of our proposed model. Qualitative and quantitative comparisons between the proposed method and the state-of-the-art methods show that the proposed method outperforms or is comparable to the state-of-the-art methods in both F1 score and precision. In particular, compared with U-Net, the mIoU of our proposed method is increased about 3%~17% compared with the three public datasets. |
format |
article |
author |
Xing Hu Minghui Yao Dawei Zhang |
author_facet |
Xing Hu Minghui Yao Dawei Zhang |
author_sort |
Xing Hu |
title |
Road crack segmentation using an attention residual U-Net with generative adversarial |
title_short |
Road crack segmentation using an attention residual U-Net with generative adversarial |
title_full |
Road crack segmentation using an attention residual U-Net with generative adversarial |
title_fullStr |
Road crack segmentation using an attention residual U-Net with generative adversarial |
title_full_unstemmed |
Road crack segmentation using an attention residual U-Net with generative adversarial |
title_sort |
road crack segmentation using an attention residual u-net with generative adversarial |
publisher |
AIMS Press |
publishDate |
2021 |
url |
https://doaj.org/article/ccc0569333544ed7a9cc36fc49156950 |
work_keys_str_mv |
AT xinghu roadcracksegmentationusinganattentionresidualunetwithgenerativeadversarial AT minghuiyao roadcracksegmentationusinganattentionresidualunetwithgenerativeadversarial AT daweizhang roadcracksegmentationusinganattentionresidualunetwithgenerativeadversarial |
_version_ |
1718407551612092416 |