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...
Autores principales: | , , |
---|---|
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 |