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 条
  • [1] Improved Small Object Detection Algorithm Based on YOLOv5
    Xu, Bo
    Gao, Bin
    Li, Yunhu
    IEEE INTELLIGENT SYSTEMS, 2024, 39 (05) : 57 - 65
  • [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] An improved lightweight object detection algorithm for YOLOv5
    Luo, Hao
    Wei, Jiangshu
    Wang, Yuchao
    Chen, Jinrong
    Li, Wujie
    PEERJ COMPUTER SCIENCE, 2024, 10
  • [4] Improved YOLOv5 for Small Object Detection Algorithm
    Yu, Jun
    Jia, Yinshan
    Computer Engineering and Applications, 2023, 59 (12) : 201 - 207
  • [5] 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
  • [6] 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
  • [7] 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
  • [8] 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
  • [9] 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
  • [10] An Improved Underwater Object Detection Algorithm Based on YOLOv5 for Blurry Images
    Cheng, Liyan
    Zhou, Hui
    Le, Xingni
    Chen, Wanru
    Tao, Hechuan
    Ding, Jiarui
    Wang, Xinru
    Wang, Ruizhi
    Yang, Qunhui
    Chen, Chen
    Kong, Meiwei
    2024 12TH INTERNATIONAL CONFERENCE ON INTELLIGENT COMPUTING AND WIRELESS OPTICAL COMMUNICATIONS, ICWOC, 2024, : 42 - 47