Multi-object detection for crowded road scene based on ML-AFP of YOLOv5

Aiming at the problem of multi-object detection such as target occlusion and tiny targets in road scenes, this paper proposes an improved YOLOv5 multi-object detection model based on ML-AFP (multi-level aggregation feature perception) mechanism. Since tiny targets such as non-motor vehicle and pedes...

Full description

Saved in:
Bibliographic Details
Published in:Scientific reports 2023-10, Vol.13 (1), p.17310-17310, Article 17310
Main Authors: Li, Yiming, Wu, Kaiwen, Kang, Wenshuo, Zhou, Yuhui, Di, Fan
Format: Article
Language:English
Subjects:
639/166/987
639/705/258
Accuracy
Deep learning
Humanities and Social Sciences
Motor vehicles
multidisciplinary
Occlusion
Pedestrians
Recall
Roads
Science
Science (multidisciplinary)
Vehicles
Citations: Items that this one cites
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Aiming at the problem of multi-object detection such as target occlusion and tiny targets in road scenes, this paper proposes an improved YOLOv5 multi-object detection model based on ML-AFP (multi-level aggregation feature perception) mechanism. Since tiny targets such as non-motor vehicle and pedestrians are not easily detected, this paper adds a micro target detection layer and a double head mechanism to improve the detection ability of tiny targets. Varifocal loss is used to achieve a more accurate ranking in the process of non-maximum suppression to solve the problem of target occlusion, and this paper also proposes a ML-AFP mechanism. The adaptive fusion of spatial feature information at different scales improves the expression ability of network model features, and improves the detection accuracy of the model as a whole. Our experimental results on multiple challenging datasets such as KITTI, BDD100K, and show that the accuracy, recall rate and mAP value of the proposed model are greatly improved, which solves the problem of multi-object detection in crowded road scenes.
ISSN:2045-2322
2045-2322
DOI:10.1038/s41598-023-43458-3