LAG-YOLO: Efficient Road Damage Detector via Lightweight Attention Ghost Module
Junxin Chen, Xiaojie Yu, Qiankun Li, Wei Wang, Ben‐Guo He
Abstract
Road damage detection plays an important role in ensuring road safety and improving traffic flow. The dramatic progress of artificial intelligence technology offers new opportunities for this field. In this paper, we introduce 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 SimAM attention mechanism. LAG- YOLO achieves an impressive parameter reduction to 4.19 million, delivering remarkable 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.