Cargando…

A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking

To solve the problems of low accuracy and false counts of existing models in road damage object detection and tracking, in this paper, we propose Road-TransTrack, a tracking model based on transformer optimization. First, using the classification network based on YOLOv5, the collected road damage im...

Descripción completa

Detalles Bibliográficos
Autores principales: Wang, Niannian, Shang, Lihang, Song, Xiaotian
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10490637/
https://www.ncbi.nlm.nih.gov/pubmed/37687850
http://dx.doi.org/10.3390/s23177395
_version_ 1785103886235729920
author Wang, Niannian
Shang, Lihang
Song, Xiaotian
author_facet Wang, Niannian
Shang, Lihang
Song, Xiaotian
author_sort Wang, Niannian
collection PubMed
description To solve the problems of low accuracy and false counts of existing models in road damage object detection and tracking, in this paper, we propose Road-TransTrack, a tracking model based on transformer optimization. First, using the classification network based on YOLOv5, the collected road damage images are classified into two categories, potholes and cracks, and made into a road damage dataset. Then, the proposed tracking model is improved with a transformer and a self-attention mechanism. Finally, the trained model is used to detect actual road videos to verify its effectiveness. The proposed tracking network shows a good detection performance with an accuracy of 91.60% and 98.59% for road cracks and potholes, respectively, and an F1 score of 0.9417 and 0.9847. The experimental results show that Road-TransTrack outperforms current conventional convolutional neural networks in terms of the detection accuracy and counting accuracy in road damage object detection and tracking tasks.
format Online
Article
Text
id pubmed-10490637
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-104906372023-09-09 A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking Wang, Niannian Shang, Lihang Song, Xiaotian Sensors (Basel) Article To solve the problems of low accuracy and false counts of existing models in road damage object detection and tracking, in this paper, we propose Road-TransTrack, a tracking model based on transformer optimization. First, using the classification network based on YOLOv5, the collected road damage images are classified into two categories, potholes and cracks, and made into a road damage dataset. Then, the proposed tracking model is improved with a transformer and a self-attention mechanism. Finally, the trained model is used to detect actual road videos to verify its effectiveness. The proposed tracking network shows a good detection performance with an accuracy of 91.60% and 98.59% for road cracks and potholes, respectively, and an F1 score of 0.9417 and 0.9847. The experimental results show that Road-TransTrack outperforms current conventional convolutional neural networks in terms of the detection accuracy and counting accuracy in road damage object detection and tracking tasks. MDPI 2023-08-24 /pmc/articles/PMC10490637/ /pubmed/37687850 http://dx.doi.org/10.3390/s23177395 Text en © 2023 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Wang, Niannian
Shang, Lihang
Song, Xiaotian
A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking
title A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking
title_full A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking
title_fullStr A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking
title_full_unstemmed A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking
title_short A Transformer-Optimized Deep Learning Network for Road Damage Detection and Tracking
title_sort transformer-optimized deep learning network for road damage detection and tracking
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10490637/
https://www.ncbi.nlm.nih.gov/pubmed/37687850
http://dx.doi.org/10.3390/s23177395
work_keys_str_mv AT wangniannian atransformeroptimizeddeeplearningnetworkforroaddamagedetectionandtracking
AT shanglihang atransformeroptimizeddeeplearningnetworkforroaddamagedetectionandtracking
AT songxiaotian atransformeroptimizeddeeplearningnetworkforroaddamagedetectionandtracking
AT wangniannian transformeroptimizeddeeplearningnetworkforroaddamagedetectionandtracking
AT shanglihang transformeroptimizeddeeplearningnetworkforroaddamagedetectionandtracking
AT songxiaotian transformeroptimizeddeeplearningnetworkforroaddamagedetectionandtracking