Object Detection for Hazardous Material Vehicles Based on Improved YOLOv5 Algorithm

被引:7
|
作者
Zhu, Pengcheng [1 ]
Chen, Bolun [1 ,2 ]
Liu, Bushi [1 ]
Qi, Zifan [1 ]
Wang, Shanshan [1 ]
Wang, Ling [1 ]
机构
[1] Huaiyin Inst Technol, Fac Comp & Software Engn, Huaian 223003, Peoples R China
[2] Univ Fribourg, Dept Phys, CH-1700 Fribourg, Switzerland
基金
中国国家自然科学基金;
关键词
hazardous material vehicles; object detection; YOLOv5; attention mechanism; NETWORK;
D O I
10.3390/electronics12051257
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Hazardous material vehicles are a non-negligible mobile source of danger in transport and pose a significant safety risk. At present, the current detection technology is well developed, but it also faces a series of challenges such as a significant amount of computational effort and unsatisfactory accuracy. To address these issues, this paper proposes a method based on YOLOv5 to improve the detection accuracy of hazardous material vehicles. The method introduces an attention module in the YOLOv5 backbone network as well as the neck network to achieve the purpose of extracting better features by assigning different weights to different parts of the feature map to suppress non-critical information. In order to enhance the fusion capability of the model under different sized feature maps, the SPPF (Spatial Pyramid Pooling-Fast) layer in the network is replaced by the SPPCSPC (Spatial Pyramid Pooling Cross Stage Partial Conv) layer. In addition, the bounding box loss function was replaced with the SIoU loss function in order to effectively speed up the bounding box regression and enhance the localization accuracy of the model. Experiments on the dataset show that the improved model has effectively improved the detection accuracy of hazardous chemical vehicles compared with the original model. Our model is of great significance for achieving traffic accident monitoring and effective emergency rescue.
引用
收藏
页数:16
相关论文
共 50 条
  • [21] Vehicle And Pedestrian Detection Algorithm Based on Improved YOLOv5
    Sun, Jiuhan
    Wang, Zhifeng
    IAENG International Journal of Computer Science, 2023, 50 (04)
  • [22] Ship Target Detection Algorithm Based on Improved YOLOv5
    Zhou, Junchi
    Jiang, Ping
    Zou, Airu
    Chen, Xinglin
    Hu, Wenwu
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2021, 9 (08)
  • [23] An Improved Distraction Behavior Detection Algorithm Based on YOLOv5
    Zhou, Keke
    Zheng, Guoqiang
    Zhai, Huihui
    Lv, Xiangshuai
    Zhang, Weizhen
    Computers, Materials and Continua, 2024, 81 (02): : 2571 - 2585
  • [24] Underwater Target Detection Algorithm Based on Improved YOLOv5
    Lei, Fei
    Tang, Feifei
    Li, Shuhan
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2022, 10 (03)
  • [25] Research on improved algorithm for helmet detection based on YOLOv5
    Chun Shan
    HongMing Liu
    Yu Yu
    Scientific Reports, 13
  • [26] Helmet wearing detection algorithm based on improved YOLOv5
    Liu, Yiping
    Jiang, Benchi
    He, Huan
    Chen, Zhijun
    Xu, Zhenfa
    SCIENTIFIC REPORTS, 2024, 14 (01)
  • [27] An Aerial Image Detection Algorithm Based on Improved YOLOv5
    Shan, Dan
    Yang, Zhi
    Wang, Xiaofeng
    Meng, Xiangdong
    Zhang, Guangwei
    SENSORS, 2024, 24 (08)
  • [28] Fabric defect detection algorithm based on improved YOLOv5
    Feng Li
    Kang Xiao
    Zhengpeng Hu
    Guozheng Zhang
    The Visual Computer, 2024, 40 : 2309 - 2324
  • [29] Stacked workpieces detection algorithm based on improved YOLOv5
    Liang, Jianan
    Zhang, Jinhua
    Kong, Ruiling
    Bian, Xingrui
    Chen, Lirong
    JOURNAL OF THE CHINESE INSTITUTE OF ENGINEERS, 2024, 47 (05) : 544 - 555
  • [30] Dense Small Object Detection Algorithm Based on Improved YOLOv5 in UAV Aerial Images
    Chen, Jiahui
    Wang, Xiaohong
    Computer Engineering and Applications, 2024, 60 (03) : 100 - 109