Vehicle and Personnel Detection in Highway Tunnels Based on Improved YOLOv5

Hong-xing PENG , Chang YUAN , Wei-ye KE , Min-jun LIANG , Yong-qiang MA

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (6) : 2453 -2461.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (6) : 2453-2461 DOI: 10.12404/j.issn.1671-1815.2302227
research-article
Vehicle and Personnel Detection in Highway Tunnels Based on Improved YOLOv5
Author information +
History +
PDF

Abstract

To address the problems of dim light, image affected by lamp light and difficulty in detecting small targets at long distance in highway tunnel, an improved YOLOv5 algorithm for vehicle and personnel detection in highway tunnel was proposed. Firstly, Gaussian mixture clustering was used to obtain a set of anchor boxes that match the dataset targets better, improving the detection accuracy of the model. Secondly, CARAFE operator was introduced in the feature fusion to enlarge the receptive field and reduce the loss of feature detail information in the upsampling process. Finally, by inserting coordinate attention (CA) into the network, the model's ability to extract features from different positions of the image is further enhanced. To verify the effectiveness of the algorithm, experiments were conducted on the Zhejiang Wenli Expressway Tunnel dataset, and the results show that: the mean average precision (mAP) of the proposed algorithm reaches 95.7%, which is 3.8% higher than the original model, and it can achieve more accurate detection for small targets at long distances and targets affected by severe lighting, providing an effective solution for vehicle and personnel detection in highway tunnels under complex environments.

Cite this article

Download citation ▾
Hong-xing PENG, Chang YUAN, Wei-ye KE, Min-jun LIANG, Yong-qiang MA. Vehicle and Personnel Detection in Highway Tunnels Based on Improved YOLOv5. Science Technology and Engineering, 2024, 24(6): 2453-2461 DOI:10.12404/j.issn.1671-1815.2302227

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (2889KB)

17

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉