Abstract: Due to the frequent accidents at the tunnel construction site, using robots instead of manual safety inspections can effectively protect the safety of staff. However, most of the robots in this field currently require manual control and lack autonomous obstacle avoidance capabilities. Aiming at the obstacle detection method in tunnel construction environment, a lightweight model using improved YOLOv5 for obstacle detection was proposed. Firstly, the obstacle dataset in the tunnel scenario was constructed. Secondly, the backbone network was modified to a lightweight Shufflenet v2 network, and the activation function was modified to the SiLU function to improve the detection speed and reduce the amount of computation. Next, the coordinate attention mechanism was incorporated to improve the network's ability to learn and represent features. Finally, the neck convolution block was modified to GSConv, which reduced the calculation amount of the model and improved the detection accuracy of the algorithm. Comparative experiments on the dataset constructed in this research showed that the detection speed of the proposed method was increased by approximately 57% compared to the original YOLOv5-n algorithm, which reduced the hardware requirements of the model.
吴江涛,李英杰. 隧道施工环境中障碍物的轻量化目标检测算法[J]. 隧道与地下工程灾害防治, 2025, 7(1): 48-56.
WU Jiangtao, LI Yingjie. The lightweight object detection algorithm for obstacles in tunnel construction environments. Hazard Control in Tunnelling and Underground Engineering, 2025, 7(1): 48-56.
[1] 郝楠. 基于深度学习的公路障碍物检测的研究[D]. 成都:电子科技大学, 2019. HAO Nan. Research on highway obstacle detection based on deep learning[D]. Chengdu: University of Electronic Science and Technology of China, 2019. [2] 郑腾龙. 基于深度学习的无人机电力巡检障碍物目标检测与避障系统[D]. 天津:天津工业大学, 2021. ZHENG Tenglong. Obstacle target detection and obstacle avoidance system for UAV power inspection based on deep learning[D]. Tianjin: Tianjin Polytechnic University, 2021. [3] 管晓勇. 基于深度学习的列车障碍物检测与识别技术研究[D]. 北京: 北京交通大学, 2020. GUAN Xiaoyong. Research on train obstacle detection and recognition technology based on deep learning[D]. Beijing: Beijing Jiaotong University, 2020. [4] 夏长权, 汪李超, 韩一帆, 等. 融合Shufflenet-v2的Yolov5轻量化目标检测方法[J]. 信息技术与信息化, 2023(3):100-104. [5] MA N N, ZHANG X Y, ZHENG H T, et al. Shufflenet v2: practical guidelines for efficient CNN architecture design[C] //Computer Vision-ECCV 2018. [S.l.] : Springer International Publishing, 2018:122-138. [6] 罗禹杰, 张剑, 陈亮, 等. 基于自适应空间特征融合的轻量化目标检测算法[J]. 激光与光电子学进展,2022,59(4): 0415004. LUO Yujie, ZHANG Jian, CHEN Liang, et al. Lightweight target detection algorithm based on adaptive spatial feature fusion[J]. Laser & Optoelectronics Progress, 2022, 59(4):0415004. [7] HOWARD A G, ZHU M L, CHEN B, et al. MobileNets: efficient convolutional neural networks for mobile vision applications[EB/OL].(2017-04-16)[2024-12-19]. https://api.semanticscholar.org/CorpusID:12670695 [8] LIU S T, HUANG D, WANG Y H. Learning spatial fusion for single-shot object detection[EB/OL].(2019-11-21)[2024-12-19]. https://arxiv.org/abs/1911.09516 [9] 杨永波, 李栋. 改进YOLOv5的轻量级安全帽佩戴检测算法[J]. 计算机工程与应用, 2022, 58(9):201-207. YANG Yongbo, LI Dong. Lightweight helmet wearing detection algorithm of improved YOLOv5[J]. Computer Engineering and Applications, 2022, 58(9):201-207. [10] HOWARD A, SANDLER M, CHEN B, et al. Searching for MobileNet v3[C] //2019 IEEE/CVF International Conference on Computer Vision. Seoul, Korea:IEEE, 2019:1314-1324. [11] GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C] //2014 IEEE Conference on Computer Vision and Pattern Recognition.Columbus, USA: IEEE, 2014:580-587. [12] GIRSHICK R. Fast R-CNN[C] //Proceedings of 2015 IEEE International Conference on Computer Vision. Santiago,Republic of Chile: IEEE, 2015:1440-1448. [13] REN S Q, HE K M, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6):1137-1149. [14] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection[C] //2016 IEEE Conference on Computer Vision and Pattern Recognition(CVPR). Las Vegas, USA:IEEE, 2016:779-788. [15] REDMON J, FARHADI A. YOLO9000: better, faster, stronger[C] //2017 IEEE Conference on Computer Vision and Pattern Recognition(CVPR). Honolulu, USA:IEEE, 2017:6517-6525. [16] 邱天衡, 王玲, 王鹏, 等. 基于改进YOLOv5的目标检测算法研究[J]. 计算机工程与应用, 2022, 58(13):63-73. QIU Tianheng, WANG Ling, WANG Peng, et al. Research on object detection algorithm based on improved YOLOv5[J]. Computer Engineering and Applications, 2022, 58(13):63-73. [17] 马琳琳, 马建新, 韩佳芳, 等. 基于YOLOv5s目标检测算法的研究[J]. 电脑知识与技术, 2021, 17(23):100-103. MA Linlin, MA Jianxin, HAN Jiafang, et al. Research on object detection algorithm based on Yolov5s[J]. Computer Knowledge and Technology, 2021, 17(23):100-103. [18] HAN K, WANG Y H, TIAN Q, et al. GhostNet: more features from cheap operations[C] //2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR). Seattle, USA:IEEE, 2020:1580-1589. [19] GLOROT X, BORDES A, BENGIO Y. Deep sparse rectifier neural networks[J]. Journal of Machine Learning Research, 2011, 15:315-323. [20] ELFWING S, UCHIBE E, DOYA K. Sigmoid-weighted linear units for neural network function approximation in reinforcement learning[J]. Neural Networks, 2018, 107:3-11. [21] HU J, SHEN L, SUN G. Squeeze-and-excitation networks[C] //2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition. Salt Lake City, USA: IEEE, 2018:7132-7141. [22] WANG Q L, WU B G, ZHU P F, et al. ECA-net: efficient channel attention for deep convolutional neural networks[C] //2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR). Seattle, USA:IEEE, 2020:11531-11539. [23] WOO S, PARK J, LEE J Y, et al. CBAM: convolutional block attention module[C] //Computer Vision-ECCV 2018. Munich, Germany: Springer International Publishing, 2018:3-19. [24] HOU Q, ZHOU D, FENG J. Coordinate attention for efficient mobile network design[C] //Proceedings of the 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR). Seattle, USA: IEEE, 2021:13713-13722. [25] LI H, LI J, WEI H, et al. Slim-neck by GSConv: a better design paradigm of detector architectures for autonomous vehicles[EB/OL]. [2024-12-19]. https://api.semanticscholar.org/CorpusID:249394561
2025, Vol. 7 
