计算机科学
行人检测
行人
目标检测
计算机视觉
人工智能
对象(语法)
模式识别(心理学)
工程类
运输工程
出处
期刊:2018 IEEE 4th Information Technology and Mechatronics Engineering Conference (ITOEC)
日期:2023-09-15
卷期号:: 260-264
标识
DOI:10.1109/itoec57671.2023.10291428
摘要
Traditional pedestrian object detection techniques, represented by YOLOv5, are limited in real-world applications due to their large parameter volumes and high computational complexity. In response to this problem, we propose an improved YOLOv5s pedestrian object detection algorithm. Firstly, we introduce the lightweight network GhostNet into the structure of YOLOv5s to achieve lightweight detection networks. On this basis, a CBAM based on the attention mechanism is proposed to enhance feature extraction and improve the performance of the model while reducing the model parameters. Secondly, we replace the CIoU loss function in the original network with the EIoU loss function, which enables the network training process to converge faster. Finally, we construct a pedestrian object detection dataset to verify the accuracy and real-time performance of the model. The experimental results show that while maintaining high accuracy, the parameter volume of the improved YOLOv5s algorithm is reduced by 45.8% compared to the original algorithm.
科研通智能强力驱动
Strongly Powered by AbleSci AI