|
|
一种基于RDNet的道路病害检测算法
|
Abstract:
道路病害的诊断是道路预防保养的一个关键步骤，为此本文提出了一种基于RDNet (Road Detection Network)道路病害检测算法。该算法从不同角度提高了特征的提取和表达能力，其中的改进包括跨阶段多分支卷积、残差并行空洞卷积以及自适应尺度空间注意力模块等。通过在自建的道路病害数据集上进行端到端地训练，提高了算法的检测精度和泛化能力。实验结果表明，对比YOLOv5s，本文所提出的RDNet算法的平均精度均值mAP提高了1.3%，同时对于困难样本也有较好的检测结果，能够有效地应用于实际道路的维护工作中，从而提升道路病害检测的效率和准确性。
The diagnosis of road diseases is a key step in road preventive maintenance, so this paper proposes a road disease detection algorithm based on RDNet (Road Detection Network). The algorithm improves the ability of feature extraction and expression from different perspectives, including crossstage partial multi-branch convolution, residual parallel dilated convolution, and adaptive scale spatial attention module. End-to-end training on the self-built road disease dataset improves the detection accuracy and generalization ability of the algorithm. Experimental results show that compared with YOLOv5s, the average precision of the RDNet algorithm proposed in this paper is increased by 1.3%, and the average precision mAP of the proposed RDNet algorithm is improved by 1.3%, and it also has good detection results for difficult samples, which can be effectively applied to the maintenance of actual roads, so as to improve the efficiency and accuracy of road disease detection.
| [1] | 罗晖, 贾晨, 李健. 基于改进 YOLOv4 的公路路面病害检测算法[J]. 激光与光电子学进展, 2021, 58(14): 328-336. |
| [2] | 沙爱民, 童峥, 高杰. 基于卷积神经网络的路表病害识别与测量[J]. 中国公路学报, 2018, 31(1): 1-10. |
| [3] | Ye, T., Zhang, X., Zhang, Y., et al. (2020) Railway Traffic Object Detection Using Differential Feature Fusion Convolution Neural Network. IEEE Transactions on Intelligent Transportation Systems, 22, 1375-1387. https://doi.org/10.1109/TITS.2020.2969993 |
| [4] | Lecun, Y., Bengio, Y. and Hinton, G. (2015) Deep Learning. Nature, 521, 436-444. https://doi.org/10.1038/nature14539 |
| [5] | 邵延华, 张铎, 楚红雨, 等. 基于深度学习的YOLO目标检测综述[J]. 电子与信息学报, 2022, 44(10): 3697-3708. |
| [6] | 张阳婷, 黄德启, 王东伟, 等. 基于深度学习的目标检测算法研究与应用综述[J]. 计算机工程与应用, 2023, 59(18): 1-13. |
| [7] | Zhao, Z.Q., Zheng, P., Xu, S., et al. (2019) Object Detection with Deep Learning: A Review. IEEE Transactions on Neural Networks and Learning Systems, 30, 3212-3232. https://doi.org/10.1109/TNNLS.2018.2876865 |
| [8] | Wang, C.Y., Liao, H.Y.M., Wu, Y.H., et al. (2020) Cspnet: A New Backbone That Can Enhance Learning Capability of CNN. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, Seattle， 14-19 June 2020, 390-391. https://doi.org/10.1109/CVPRW50498.2020.00203 |
| [9] | He, K., Zhang, X., Ren, S., et al. (2016) Deep Residual Learning for Image Recognition. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, 27-30 June 2016, 770-778. https://doi.org/10.1109/CVPR.2016.90 |
| [10] | Hou, Q., Zhou, D. and Feng, J. (2021) Coordinate Attention for Efficient Mobile Network Design. Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Nashville, 20-25 June 2021, 13713-13722. https://doi.org/10.1109/CVPR46437.2021.01350 |
| [11] | Zhang, Q.L. and Yang, Y.B. (2021) Sa-Net: Shuffle Attention for Deep Convolutional Neural Networks. ICASSP 2021-2021 IEEE International Conference on Acoustics, Speech and Signal Processing, Toronto, 6-11 June 2021, 2235-2239. https://doi.org/10.1109/ICASSP39728.2021.9414568 |
| [12] | Zheng, Z., Wang, P., Liu, W., et al. (2020) Distance-Iou Loss: Faster and Better Learning for Bounding Box Regression. Proceedings of the AAAI Conference on Artificial Intelligence, 34, 12993-13000. https://doi.org/10.1609/aaai.v34i07.6999 |
| [13] | Rezatofighi, H., Tsoi, N., Gwak, J.Y., et al. Generalized Intersection Over Union: A Metric and A Loss For Bounding Box Regression. 2019 Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, Long Beach, 15-20 June 2019, 658-666. https://doi.org/10.1109/CVPR.2019.00075 |
| [14] | Zhou, D., Fang, J., Song, X., et al. (2019) Iou Loss for 2d/3d Object Detection. 2019 International Conference on 3D Vision, Québec, 16-19 September 2019, 85-94. https://doi.org/10.1109/3DV.2019.00019 |
| [15] | Jocher, G., Chaurasia, A., Stoken, A., et al. (2022) Ultralytics/Yolov5: V6. 2-Yolov5 Classification Models, Apple M1, Reproducibility, Clearml and Deci. Ai Integrations. |
