A Lightweight Method for Road Damage Detection Based on Improved YOLOv8n

被引:0
作者
Li, Xudong [1 ]
Zhang, Yujun [1 ]
机构
[1] Univ Sci & Technol Liaoning, Sch Comp & Software Engn, Anshan 114051, Peoples R China
关键词
Index Terms; road damage detection; lightweight;
D O I
暂无
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
substantial challenges in striking an optimal balance between precision and processing speed. Furthermore, owing to the extensive number of parameters, these models present significant challenges for effective deployment on edge devices with constrained computational resources. In order to tackle these challenges, this paper introduces a lightweight road damage detection model, BSE-YOLO, which is founded on the optimization of the YOLOv8n architecture. Initially, we reformulate the feature fusion network by integrating the concept of BiFPN to minimize both the parameter count and computational overhead. Subsequently, a novel SC2f module is introduced and integrated into the feature fusion network, thereby further minimizing both the parameter count and computational requirements. Additionally, this study presents the SEAHead module, which makes use of limited computational resources to obtain vital information, consequently improving both the efficiency and precision of detection tasks while reducing computational costs. The experimental results indicate that, in comparison to the original model, the BSEYOLO algorithm achieves a 40% reduction in parameters, a decrease of 2.2G in FLOPs, an increase of 3 in FPS, and only a 0.1% decline in mAP@0.5. This model effectively fulfills the accuracy and real-time processing demands for road damage detection tasks on mobile edge devices.
引用
收藏
页码:114 / 123
页数:10
相关论文
共 37 条
[1]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, DOI 10.48550/ARXIV.2004.10934]
[2]  
Chatterjee A, 2018, EUR SIGNAL PR CONF, P2140, DOI 10.23919/EUSIPCO.2018.8553388
[3]   Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks [J].
Chen, Jierun ;
Kao, Shiu-Hong ;
He, Hao ;
Zhuo, Weipeng ;
Wen, Song ;
Lee, Chul-Ho ;
Chan, S. -H. Gary .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, :12021-12031
[4]  
Daniel A., 2014, International Journal of Computer Applications, V101
[5]   Detection of Crack Growth in Asphalt Pavement Through Use of Infrared Imaging [J].
Du, Yuchuan ;
Zhang, Xiaoming ;
Li, Feng ;
Sun, Lijun .
TRANSPORTATION RESEARCH RECORD, 2017, (2645) :24-31
[6]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[7]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587
[8]  
Glenn J., 2022, Yolov5 release v6. 1, V1
[9]   Road damage detection algorithm for improved YOLOv5 [J].
Guo, Gege ;
Zhang, Zhenyu .
SCIENTIFIC REPORTS, 2022, 12 (01)
[10]   Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) :1904-1916