LAG-YOLO: Efficient road damage detector via lightweight attention ghost module
| Autor: | Junxin Chen, Xiaojie Yu, Qiankun Li, Wei Wang, Ben-Guo He |
|---|---|
| Jazyk: | angličtina |
| Rok vydání: | 2024 |
| Předmět: | |
| Zdroj: | Journal of Intelligent Construction, Vol 2, Iss 1, p 9180032 (2024) |
| Druh dokumentu: | article |
| ISSN: | 2958-3861 2958-2652 |
| DOI: | 10.26599/JIC.2023.9180032 |
| Popis: | Road damage detection plays an important role in ensuring road safety and improving traffic flow. The dramatic progress of artificial intelligence (AI) technology offers new opportunities for this field. In this paper, we introduce lightweight attention ghost-you only look once (LAG-YOLO), an efficient deep-learning network for road damage detection. LAG-YOLO optimizes the network structure of YOLO, making it more suitable for real-time processing and lightweight deployment while ensuring high accuracy. In addition, a novel module called attention ghost is designed to reduce the model parameters and improve the model performance by the simple attention module (SimAM). LAG-YOLO achieves an impressive parameter reduction to 4.19 million, delivering remarkable mean average precision (mAP) scores of 45.80% on the Hualu dataset and 52.35% on the RDD2020 dataset. In summary, the proposed network performs satisfactorily on extensive road damage datasets with fewer parameters, making it more suitable to be deployed in practice. |
| Databáze: | Directory of Open Access Journals |
| Externí odkaz: |
|