首页|适用于边缘设备的轻量级人体检测算法

适用于边缘设备的轻量级人体检测算法

扫码查看
针对现存的人体检测网络都比较复杂,部署到边缘设备上时表现不佳的问题,基于YOLOv7提出一种轻量级人体检测算法.该算法首先使用改进后的ShuffleNev2基本模块替换原网络ELAN模块;接着在主干网络末端添加SE注意力和SPPF池化;然后在Neck部分使用改进后的GSConv替换标准卷积,引入基于GSConv的VoVGSCSP替换ELAN-W模块.通过在GPU和Sophon SE5上的验证结果表明,该轻量级人体检测算法与YOLOv7相比损失2.6%的精度,但计算量大幅度降低,在Sophon SE5上推理速度达到了54 FPS,相比较YOLOv7提升了39 FPS.
Lightweight human body detection algorithm suitable for edge devices
A lightweight human detection algorithm based on YOLOv7 is proposed to address the issue of complex human de-tection networks that perform poorly when deployed on edge devices.The algorithm first replaces the original network ELAN mod-ule with the improved ShuffleNev2 basic module;Next,add SE attention and SPPF pooling at the end of the backbone network;Then,in the Neck section,the improved GSConv is used to replace the standard convolution,and the GSConv based VoVGSCSP is introduced to replace the ELAN-W module.The validation results on GPU and Sophon SE5 show that this lightweight human detec-tion algorithm loses 2.6%accuracy compared to YOLOv7,but significantly reduces computational complexity.The inference speed on Sophon SE5 reaches 54 FPS,which is 39 FPS higher than YOLOv7.

YOLOv7target detectionShuffleNetedge computingSophon SE5

周宁、陶青川、彭勃兴

展开 >

四川大学电子信息学院,成都 610065

YOLOv7 目标检测 ShuffleNet 边缘计算 Sophon SE5

2024

现代计算机
中大控股

现代计算机

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