IVP-YOLOv5: an intelligent vehicle-pedestrian detection method based on YOLOv5s

被引:7
|
作者
Sun, Yang [1 ,2 ]
Song, Jiankun [1 ]
Li, Yong [3 ]
Li, Yi [1 ]
Li, Song [1 ]
Duan, Zehao [1 ]
机构
[1] Hebei Univ Engn, Sch Mech & Equipment Engn, Handan, Peoples R China
[2] Handan Key Lab Intelligent Vehicles, Handan, Peoples R China
[3] Handan Hansan Construct Engn Co Ltd, Handan, Peoples R China
关键词
YOLOv5s; pedestrian detection; Ghost-BottleNeck; Alpha-IoU; SAHI; OBJECT DETECTION;
D O I
10.1080/09540091.2023.2168254
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Computer vision is now vital in intelligent vehicle environment perception systems. However, real-time small-scale pedestrian detection in intelligent vehicle environment perception systems is still needs to be improved. This paper proposes an intelligent vehicle-pedestrian detection method based on YOLOv5s, named IVP-YOLOv5, to use in vehicle environment perception systems. Based on the network structure of YOLOv5s, we replaced BottleNeck CSP with Ghost-Bottleneck to reduce the complexity of processing feature maps while maintaining good detection performance. To reduce the error between the ground truth box and the predicted box, we apply Alpha-IoU as the bounding box loss function, improving pedestrian detection accuracy and robustness. We introduce the slicing-aided hyper inference (SAHI) strategy, which enables the lightweight backbone network to capture more detailed features of pedestrians by enlarging image pixels. Experiments on the BDD100 K dataset show that the proposed IVP-YOLOv5 achieves 67.1% AP and 18.5% APs of pedestrian detection, and the GFLOPs and the number of parameters are only 10.5 and 4.9M.
引用
收藏
页数:20
相关论文
共 50 条
  • [1] Improved Pedestrian Detection Algorithm Based on YOLOv5s
    Li, Zhihua
    Zhang, Yuanbiao
    Wang, Chao
    Tan, Guopeng
    Yan, Yahui
    JOURNAL OF ADVANCED COMPUTATIONAL INTELLIGENCE AND INTELLIGENT INFORMATICS, 2024, 28 (04) : 768 - 775
  • [2] Research on Vehicle Detection Algorithms Based on YOLOv5s
    Zhang, Lifeng
    Tian, Ying
    IAENG International Journal of Computer Science, 2023, 50 (04)
  • [3] Lightweight Vehicle Detection Based on Improved YOLOv5s
    Wang, Yuhai
    Xu, Shuobo
    Wang, Peng
    Li, Kefeng
    Song, Ze
    Zheng, Quanfeng
    Li, Yanshun
    He, Qiang
    SENSORS, 2024, 24 (04)
  • [4] Pedestrian Detection and Feedback Application Based on YOLOv5s and DeepSORT
    Ling, Li
    Tao, Jun
    Wu, Gui
    2022 34TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC, 2022, : 5716 - 5721
  • [5] Vehicle-Pedestrian Detection Method Based on Improved YOLOv8
    Wang, Bo
    Li, Yuan-Yuan
    Xu, Weijie
    Wang, Huawei
    Hu, Li
    ELECTRONICS, 2024, 13 (11)
  • [6] An Enhanced YOLOv5 Model with Attention Module for Vehicle-Pedestrian Detection
    Chen, Yicheng
    Liu, Wenbo
    Zhang, Jiyong
    2022 IEEE 31ST INTERNATIONAL SYMPOSIUM ON INDUSTRIAL ELECTRONICS (ISIE), 2022, : 1035 - 1040
  • [7] A Method of Pneumonia Detection Based on an Improved YOLOv5s
    Shan, Ruiqing
    Zhang, Xiaoxia
    Li, Shicheng
    ENGINEERING LETTERS, 2024, 32 (06) : 1243 - 1254
  • [8] Research on Road Vehicle Detection based on improved Yolov5s
    Shao, Lei
    Fan, Zhenqiang
    Li, Ji
    Liu, Hongli
    PROCEEDINGS OF 2022 IEEE INTERNATIONAL CONFERENCE ON MECHATRONICS AND AUTOMATION (IEEE ICMA 2022), 2022, : 428 - 433
  • [9] Improved YOLOv5s Model for Vehicle Detection and Recognition
    Lu, Xingmin
    Song, Wei
    INTELLIGENT COMPUTING METHODOLOGIES, PT III, 2022, 13395 : 423 - 434
  • [10] Detection method of potato leaf disease based on YOLOv5s
    Li, Jingtao
    Chen, Hao
    Li, Guisong
    Liu, Yueqi
    Yang, Yanli
    Liu, Xia
    Wang, Chang Yi
    JOURNAL OF AGRICULTURAL ENGINEERING, 2024, 55 (03)