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 条
  • [1] Improved Small Object Detection Algorithm Based on YOLOv5
    Bo X.
    Bin G.
    Yunhu L.
    IEEE Intelligent Systems, 2024, 39 (05) : 1 - 9
  • [2] A Small Object Detection Algorithm Based on Improved YOLOv5
    Guo L.
    Wang Q.
    Xue W.
    Guo J.
    Dianzi Keji Daxue Xuebao/Journal of the University of Electronic Science and Technology of China, 2022, 51 (02): : 251 - 258
  • [3] Improved YOLOv5 for Small Object Detection Algorithm
    Yu, Jun
    Jia, Yinshan
    Computer Engineering and Applications, 2023, 59 (12) : 201 - 207
  • [4] Road object detection method based on improved YOLOv5 algorithm
    Wang, Hong-Zhi
    Song, Ming-Xuan
    Cheng, Chao
    Xie, Dong-Xuan
    Jilin Daxue Xuebao (Gongxueban)/Journal of Jilin University (Engineering and Technology Edition), 2024, 54 (09): : 2658 - 2667
  • [5] Lightweight object detection algorithm based on YOLOv5 for unmanned surface vehicles
    Zhang, Jialin
    Jin, Jiucai
    Ma, Yi
    Ren, Peng
    FRONTIERS IN MARINE SCIENCE, 2023, 9
  • [6] Lightweight object detection algorithm for robots with improved YOLOv5
    Liu, Gang
    Hu, Yanxin
    Chen, Zhiyu
    Guo, Jianwei
    Ni, Peng
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2023, 123
  • [7] Object Detection Algorithm of Transmission Lines Based on Improved YOLOv5 Framework
    Zhang, Hao
    Zhou, Xianjun
    Shi, Yike
    Guo, Xuan
    Liu, Hang
    JOURNAL OF SENSORS, 2024, 2024
  • [8] Small Object Detection Algorithm Based on Improved YOLOv5 in UAV Image
    Xie, Chunhui
    Wu, Jinming
    Xu, Huaiyu
    Computer Engineering and Applications, 2023, 59 (09) : 198 - 206
  • [9] Object Detection Algorithm for Fish Eye Image Based on Improved YOLOv5
    Han, Yanfeng
    Ren, Qi
    Xiao, Ke
    Hunan Daxue Xuebao/Journal of Hunan University Natural Sciences, 2024, 51 (06): : 29 - 39
  • [10] Improved YOLOv5 Algorithm for Oriented Object Detection of Aerial Image
    Yang, Gang
    Wang, Miao
    Zhou, Quan
    Li, Jiangchuan
    Zhou, Siyue
    Lu, Yutong
    SENSORS AND MATERIALS, 2024, 36 (07) : 3059 - 3073