Real-time edge AI algorithm for road defects detection
Source abstract
Abstract
Vision-based automated road defects detection schemes are significant for highway maintenance and road condition grade assessment. This paper was dedicated to achieving a refined trade-off between accuracy and efficiency through neural network optimisation. Specifically, we proposed a novel lightweight model, RT-RDD (Real-Time Road Defects Detection), which integrated low-level extraction and high-level semantic feature extraction for road defects detection. This was achieved by designing a more advantageous backbone structure and a lighter neck structure. Additionally, key improvements include optimising the feature extraction strategy and evaluation. The RT-RDD model surpasses existing models in terms of mean Average Precision (mAP), with only a marginal reduction after quantisation. Compared to foundation model, our algorithm improves mAP (0.5:0.95) by 4.2%. In addition, we propose a quantisation and compression strategy that effectively reduces the overall model size by approximately one-third. Notably, this reduction in size only results in a minor decrease of 1.3% in mAP (0.5:0.95). Testing on NVIDIA Jetson Xavier NX Development Board Kit demonstrates the model achieves a detection speed of 58 frames per second (FPS) with enhanced precision. These improvements make it ideal for deployment in practical road maintenance scenarios.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Infrastructure Maintenance and Monitoring · Industrial Vision Systems and Defect Detection · Advanced Neural Network Applications
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Nanjing Forestry University
McMaster University
References (51)
Show source-provided references
1. 10.1109/BigData50022.2020.937779010.1109/BigData50022.2020.9377790 ↗
2. 10.1155/2022/922121110.1155/2022/9221211 ↗
3. Bochkovskiy A. Wang C.-Y. & Liao H.-Y. M. (2020). Yolov4: Optimal speed and accuracy of object detection.” arXiv preprint arXiv:2004.10934.
4. Chen N. Li Q. Ning J. Wang Q. & Liao N. (2022). Improving the lightweight object detection method for YOLOv5. Paper presented at the International Workshop of Advanced Manufacturing and Automation.10.1007/978-981-19-9338-1_4 ↗
5. 10.3934/mbe.201940210.3934/mbe.2019402 ↗
6. Cui C. Gao T. Wei S. Du Y. Guo R. Dong S. Lu B. Zhou Y. Lv X. & Liu Q. (2021). PP-LCNet: A lightweight CPU convolutional neural network.” arXiv preprint arXiv:2109.15099.
7. 10.1016/j.ins.2023.03.03810.1016/j.ins.2023.03.038 ↗
8. 10.37936/ecti-cit.2022164.24867410.37936/ecti-cit.2022164.248674 ↗
9. 10.1109/CVPR.2019.0082810.1109/CVPR.2019.00828 ↗
10. Ge Z. Liu S. Wang F. Li Z. & Sun J. (2021). Yolox: Exceeding yolo series in 2021.” arXiv preprint arXiv:2107.08430.
11. 10.1109/CVPR.2019.0072010.1109/CVPR.2019.00720 ↗
12. 10.1109/ICCV.2015.16910.1109/ICCV.2015.169 ↗
13. Han, K., Xiao, A., Wu, E., Guo, J., Xu, C., & Wang, Y. (2021). Transformer in transformer. Advances in Neural Information Processing Systems, 34, 15908–15919.
14. 10.1016/j.measurement.2022.11127710.1016/j.measurement.2022.111277 ↗
15. 10.1109/ICCV.2019.0014010.1109/ICCV.2019.00140 ↗
16. Howard A. G. Zhu M. Chen B. Kalenichenko D. Wang W. Weyand T. Andreetto M. & Adam H. (2017). Mobilenets: Efficient convolutional neural networks for mobile vision applications.” arXiv preprint arXiv:1704.04861.
17. 10.1109/CVPR.2018.0074510.1109/CVPR.2018.00745 ↗
18. Intel. (2023). oneDNN: Deep Learning Performance Acceleration Library Documentation. https://oneapi-src.github.io/oneDNN.
19. 10.1109/CVPR.2018.0028610.1109/CVPR.2018.00286 ↗
20. 10.1016/j.procs.2022.01.13510.1016/j.procs.2022.01.135 ↗
21. Jocher G. Stoken A. Borovec J. Changyu L. Hogan A. Diaconu L. Ingham F. Poznanski J. Fang J. & Yu L. (2020). “ultralytics/yolov5: v3. 1-bug fixes and performance improvements.” Zenodo.
22. Ke, J., Wang, Q., Wang, Y., Milanfar, P., Yang, F. (2021). Musiq: Multi-scale image quality transformer. Proceedings of the IEEE/CVF International Conference on Computer Vision, 5148–5157.
23. 10.1016/j.aquaeng.2022.10227310.1016/j.aquaeng.2022.102273 ↗
24. 10.1109/TNNLS.2021.308482710.1109/TNNLS.2021.3084827 ↗
25. 10.1109/IJCNN54540.2023.1019149710.1109/IJCNN54540.2023.10191497 ↗
26. 10.3390/app1108372510.3390/app11083725 ↗
27. Lin T.-Y. Maire M. Belongie S. Hays J. Perona P. Ramanan D. Dollár P. & Zitnick C. L. (2014). Microsoft coco: Common objects in context. Paper presented at the Computer Vision–ECCV 2014: 13th European Conference Zurich Switzerland September 6-12 2014 Proceedings Part V 13.10.1007/978-3-319-10602-1_48 ↗
28. Liu W. Anguelov D. Erhan D. Szegedy C. Reed S. Fu C.-Y. & Berg A. C. (2016). Ssd: Single shot multibox detector. Paper presented at the Computer Vision–ECCV 2016: 14th European Conference Amsterdam The Netherlands October 11–14 2016 Proceedings Part I 14.
29. 10.1109/BigData50022.2020.937799110.1109/BigData50022.2020.9377991 ↗
30. 10.3390/f1303038310.3390/f13030383 ↗
31. 10.3390/electronics1212258310.3390/electronics12122583 ↗
32. Ma, Y., Yu, D., Wu, T., & Wang, H. (2019). PaddlePaddle: An open-source deep learning platform from industrial practice. Frontiers of Data and Domputing, 1(1), 105–115.
33. 10.1016/j.compag.2022.10751710.1016/j.compag.2022.107517 ↗
34. Redmon J. & Farhadi A. (2018). Yolov3: An incremental improvement.” arXiv preprint arXiv:1804.02767.
35. 10.1109/TPAMI.2016.257703110.1109/TPAMI.2016.2577031 ↗
36. Rezatofighi, H., Tsoi, N., Gwak, J. Y., Sadeghian, A., Reid, I., Savarese, S. (2019). Generalized intersection over union: A metric and a loss for bounding box regression. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 658–666.
37. Rodriguez, A., Segal, E., Meiri, E., Fomenko, E., Jim Kim, Y., Shen, H., & Ziv, B. (2018). Lower numerical precision deep learning inference and training. Intel White Paper, 3(1), 1–19.
38. Rukundo O. & Cao H. (2012). Nearest neighbor value interpolation.” arXiv preprint arXiv:1211.1768.
39. 10.28932/jutisi.v6i2.268810.28932/jutisi.v6i2.2688 ↗
40. 10.1109/IAEAC54830.2022.993010110.1109/IAEAC54830.2022.9930101 ↗
41. Targ S. Almeida D. & Lyman K. (2016). Resnet in resnet: Generalizing residual architectures.” arXiv preprint arXiv:1603.08029.
42. 10.1109/ICCV.2019.0031010.1109/ICCV.2019.00310 ↗
43. 10.1109/CVPRW50498.2020.0020310.1109/CVPRW50498.2020.00203 ↗
44. Wang Y. Zhou Q. Xiong J. Wu X. & Jin X. (2019b). ESNet: An efficient symmetric network for real-time semantic segmentation. Paper presented at the Pattern Recognition and Computer Vision: Second Chinese Conference PRCV 2019 Xi’an People’s Republic of China November 8–11 2019 Proceedings Part II 2.
45. 10.1109/ICCV.2017.19310.1109/ICCV.2017.193 ↗
46. Yu G. Chang Q. Lv W. Xu C. Cui C. Ji W. Dang Q. Deng K. Wang G. & Du Y. (2021). PP-PicoDet: A better real-time object detector on mobile devices.” arXiv preprint arXiv:2111.00902.
47. 10.1609/aaai.v34i07.699910.1609/aaai.v34i07.6999 ↗
48. 10.1109/TCYB.2021.309530510.1109/TCYB.2021.3095305 ↗
49. 10.3390/su1410630610.3390/su14106306 ↗
50. 10.1080/17499518.2023.217218710.1080/17499518.2023.2172187 ↗
51. 10.1016/j.undsp.2022.07.00310.1016/j.undsp.2022.07.003 ↗