改进YOLO轻量化网络的行人检测算法

针对当前行人检测方法计算量大、检测精度低的问题,基于YOLOv4-tiny提出一种改进的行人检测算法.引入通道注意力和空间注意力模块(CBAM)至CSPDarknet53-tiny网络中,通过学习图像的位置信息和通道信息得到更加丰富的特征;在骨干网络CSPDarknet53-tiny之后引入空间金字塔池化模块,能够极大...

Pedestrian detection algorithm based on improved YOLO lightweight network

As the current pedestrian detection method has the problems of large computation and low detection accuracy,an improved pedestrian detection method based on YOLOv4-Tiny was proposed.This method introduces Convolutional Block Attention Module into CSPDarknet53-tiny network to get richer features by learning the position information and channel information of the image,adds the spatial pyramid pooling module following CSPDarknet53-tiny,which can greatly increase the receptive field and isolate the most significant context features,and uses CIoU loss function to optimize the combined loss of multiple tasks.In the experiment,the training set in INRIA and WiderPerson are used as the training model,and the test set in INRIA and WiderPerson are used as the test set to verify the model.Compared with YOLOv4-Tiny,the precision,recall and average accuracy of the improved YOLOv4-Tiny network in INRIA test set are increased by 6.23%,3.15% and 6.12%,respectively,and the improved network increased the precision,recall,and average accuracy in the WiderPerson test set by 3.65%,3.28%,and 4.41%,respectively.It is found that this improved model can extract pedestrian features more easily and improve the detection accuracy on the premise that the real-time detection is hardly affected.