首页|基于改进YOLOv8的轻量化小尺度行人和非机动车目标检测算法

基于改进YOLOv8的轻量化小尺度行人和非机动车目标检测算法

扫码查看
为解决交通路口行人和非机动车治理问题,针对全景路口监控下行人和非机动车目标小等问题,提出了一种基于改进YOLOv8的轻量化小尺度行人和非机动车目标检测算法——ACM-YOLO.首先,提出轻量化的高效AFPN特征融合网络,替换原始网络中的PAFPN算法,提高小目标的识别效果;其次,基于PConv提出了CWPConv通道权重部分卷积,并进一步提出了CWPC2f,有效降低了模型参数量和计算量;最后,使用MPDIoU函数优化网络的边界框损失.结果表明,相较于YOLOv8m,mAP50在自建数据集、VisDrone2019数据集和CityPerson数据集上分别提高了2.2%、8.1%和3%,参数量下降了17%,GFLOPs下降2.
A lightweight small-scale pedestrian and non-motorised vehicle target detection algorithm based on improved YOLOv8
To solve the problem of pedestrian and non-motorized vehicle governance at traffic intersections,a small and light-weight small-scale pedestrian and non-motorized vehicle detection algorithm based on improved YOLOv8,ACM-YOLO is pro-posed,for the small size of pedestrian and non-motorized vehicle targets at panoramic intersection monitoring.Firstly,a lightweight and efficient AFPN feature fusion network is proposed to replace the PAFPN algorithm in the original network to improve the recog-nition effect of small targets;secondly,CWPConv channel weight partial convolution is proposed based on PConv,and CWPC2f is further proposed to effectively reduce the number of model parameters and calculation;finally,the MPDIoU function is used to opti-mize the boundary box loss of the network.The results show that compared with YOLOv8m,the mAP50 is increased by 2.2%,8.1%and 3%on the self-built dataset,VisDrone2019 dataset and CityPerson dataset respectively,the number of parameters is re-duced by 17%and the GFLOPs is reduced by 2.

small target detectionYOLOv8pedestrians and non-motor vehicles detectionCWPConv

帅勃宇、张雅丽

展开 >

中国人民公安大学信息网络安全学院,北京 100038

小目标检测 YOLOv8 行人和非机动车检测 CWPConv

2024

现代计算机
中大控股

现代计算机

影响因子:0.292
ISSN:1007-1423
年,卷(期):2024.30(20)