Research on road Damage Detection Based on Improved YOLOv8
DOI:
https://doi.org/10.54097/vnjm0m76Keywords:
Road detection; Deep learning;YOLOv8.Abstract
Aiming at the low accuracy of deep learning in road damage detection, a road damage detection method based on improved YOLOv8 is proposed. Firstly, SENetv2 attention module was added to the backbone network of YOLOv8 to improve the model's feature learning ability. Secondly, in the neck network, GSConv is used to replace the common convolutional module to reduce the complexity of the model and improve the accuracy. Finally, the loss function is changed to Wise-iou to reduce the impact of detection due to a small number of low-quality instances. The experimental results show that compared with the traditional YOLOv8, the mAP50 of this model is increased by 2.1 percentage points, and the detection effect is good, which can meet the requirements of accurate detection.
Downloads
References
Zhang, L., Yang, F., Zhang, Y. D., et al. (2016). Road crack detection using deep convolutional neural network. In 2016 IEEE International Conference on Image Processing (ICIP) (pp. 3708-3712). IEEE.
Wang, Q., Mao, J., Zhai, X., et al. (2021). Improvements of YoloV3 for road damage detection. In Journal of Physics: Conference Series (Vol. 1903, No. 1, p. 012008). IOP Publishing.
Wan, F., Sun, C., He, H., et al. (2022). YOLO-LRDD: a lightweight method for road damage detection based on improved YOLOv5s. EURASIP Journal on Advances in Signal Processing, 2022(1), 98.
Cha, Y. J., Choi, W., & Büyüköztürk, O. (2017). Deep learning-based crack damage detection using convolutional neural networks. Computer-Aided Civil and Infrastructure Engineering, 32(5), 361-378.
Nie, M., & Wang, K. (2018). Pavement distress detection based on transfer learning. In 2018 5th International Conference on Systems and Informatics (ICSAI) (pp. 435-439). IEEE.
Yang, D. M., Zhang, J. G., Xu, S. B., et al. (2018). Real-time pedestrian detection via hierarchical convolutional feature. Multimedia Tools and Applications, 77(19), 25841-25860.
Sun, Z., Pei, L., Li, W., et al. (2020). Pavement sealed crack detection method based on improved Faster R-CNN. Journal of South China University of Technology (Natural Science Edition), 48(02), 84-93.
Gu, S., Li, X., Wang, X., et al. (2021). Crack detection based on enhanced semantic information and multi-channel feature fusion. Computer Engineering and Applications, 57(10), 204-210.
Narayanan, M. (2023). SENetV2: Aggregated dense layer for channelwise and global representations. arXiv preprint arXiv:2311.10807.
Li, H., Li, J., Wei, H., et al. (2022). Slim-neck by GSConv: A better design paradigm of detector architectures for autonomous vehicles. arXiv preprint arXiv:2206.02424.
Tong, Z., Chen, Y., Xu, Z., et al. (2023). Wise-IoU: Bounding Box Regression Loss with Dynamic Focusing Mechanism. arXiv preprint arXiv:2301.10051.
Arya, D., Maeda, H., Ghosh, S. K., et al. (2022). Crowdsensing-based Road Damage Detection Challenge (CRDDC’2022). In 2022 IEEE International Conference on Big Data (Big Data) (pp. 6378-6386). IEEE.
Downloads
Published
Issue
Section
License
Copyright (c) 2024 Academic Journal of Science and Technology
This work is licensed under a Creative Commons Attribution 4.0 International License.