Object Detection for Hazardous Material Vehicles Based on Improved YOLOv5 Algorithm

被引:10
|
作者
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 条
  • [31] Underwater Target Detection Algorithm Based on Improved YOLOv5
    Lei, Fei
    Tang, Feifei
    Li, Shuhan
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2022, 10 (03)
  • [32] An Aerial Image Detection Algorithm Based on Improved YOLOv5
    Shan, Dan
    Yang, Zhi
    Wang, Xiaofeng
    Meng, Xiangdong
    Zhang, Guangwei
    SENSORS, 2024, 24 (08)
  • [33] Helmet wearing detection algorithm based on improved YOLOv5
    Liu, Yiping
    Jiang, Benchi
    He, Huan
    Chen, Zhijun
    Xu, Zhenfa
    SCIENTIFIC REPORTS, 2024, 14 (01)
  • [34] 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
  • [35] Improved Plate Defect Detection Algorithm Based on YOLOv5
    Wang, Zijie
    Wang, Lan
    Zheng, Sihui
    IOT AS A SERVICE, IOTAAS 2023, 2025, 585 : 371 - 384
  • [36] Fabric defect detection algorithm based on improved YOLOv5
    Feng Li
    Kang Xiao
    Zhengpeng Hu
    Guozheng Zhang
    The Visual Computer, 2024, 40 : 2309 - 2324
  • [37] Small Target Detection Algorithm Based on Improved YOLOv5
    Chen, Ruiyun
    Liu, Zhonghua
    Ou, Weihua
    Zhang, Kaibing
    ELECTRONICS, 2024, 13 (21)
  • [38] Real⁃time dense small object detection algorithm for UAV based on improved YOLOv5
    Feng Z.
    Xie Z.
    Bao Z.
    Chen K.
    Hangkong Xuebao/Acta Aeronautica et Astronautica Sinica, 2023, 44 (07):
  • [39] 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
  • [40] Insulator defect detection based on improved YOLOv5 algorithm
    Wang, Yongheng
    Li, Qin
    Liu, Yachong
    Wang, Chao
    2023 IEEE 12TH DATA DRIVEN CONTROL AND LEARNING SYSTEMS CONFERENCE, DDCLS, 2023, : 770 - 775