Issue |
MATEC Web Conf.
Volume 336, 2021
2020 2nd International Conference on Computer Science Communication and Network Security (CSCNS2020)
|
|
---|---|---|
Article Number | 06002 | |
Number of page(s) | 8 | |
Section | Artificial Recognition and Application | |
DOI | https://doi.org/10.1051/matecconf/202133606002 | |
Published online | 15 February 2021 |
Research on pedestrian detection algorithm in driverless urban traffic environment
Tianjin University of Technology and Education, Tianjin 300222, China
* Corresponding author: liuxinchao_com@sina.com
Pedestrian detection in urban traffic environment is an important field of driverless vehicle research. Due to the variability of traffic flow, target detection algorithm cannot extract complete feature information, which brings great challenges to driverless pedestrian detection. Target detection algorithm YOLOv4 has excellent detection performance in object detection, but it is not perfect in identifying semi-blocked pedestrians. In this paper, the Spatial Pyramid Pooling was added in front of the third yolo detection head module of YOLOv4 to optimize the extraction of deep network features. Then, on the basis of optimizing the network, pruning strategy was adopted to simplify the target detection algorithm, which was called TidyYOLOv4.TidyYOLOv4 and YOLOv4 (network set input image size is 864×864) were compared on the self-made human head data set. Total BFLOPS decreased by 95.04% and Inference time decreased by 82.82%. The above experimental results show that the optimized TidyYOLOv4 algorithm is more suitable for driverless pedestrian detection in urban traffic environment.
Key words: TidyYOLOv4 / Pedestrian detection / Model pruning / Unmanned
© The Authors, published by EDP Sciences, 2021
This is an Open Access article distributed under the terms of the Creative Commons Attribution License 4.0, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
Current usage metrics show cumulative count of Article Views (full-text article views including HTML views, PDF and ePub downloads, according to the available data) and Abstracts Views on Vision4Press platform.
Data correspond to usage on the plateform after 2015. The current usage metrics is available 48-96 hours after online publication and is updated daily on week days.
Initial download of the metrics may take a while.