首页|改进YOLOv8的轻量化猪脸关键点检测

改进YOLOv8的轻量化猪脸关键点检测

扫码查看
原文链接
  • NETL
  • NSTL
  • 万方数据
为了解决 YOLOv8-pose关键点检测网络结构复杂、计算量多、参数多等问题,提出一种基于YOLOv8-pose改进的轻量化模型 SGS-YOLOv8-pose.在原有的 YOLOv8-pose 网络上替换骨干网络为 ShuffleNetv2,将 GhostConv模块整合到 Neck中,降低网络计算量,在网络层中加入 SE注意力机制,提高模型计算精度;同时采用了 WIOU损失函数,使模型的收敛效果更好;改进后的模型与原始模型相比,目标框检测精度提升 0.30%,浮点计算量减少 33.70%,参数量减少 38.55%,模型大小降低了 37.18%.试验结果表明:在不同轻量化网络与特征融合方式下,在不同的注意力机制下作对比下,改进后的 SGS-YOLOv8-pose模型,能够在保持较高检测精度下使模型浮点计算量和参数量大幅降低,便于后期模型的部署.
Improved Yolov8 for Lightweight Pen Face Keypoints Detection
In order to solve the problems of complex structure,high computational complexity,and multiple parameters in the YOLOv8-pose keypoints detection network,an improved lightweight model SGS-YOLOv8-pose based on YOLOv8-pose is proposed.It replaces the backbone network with ShuffleNetv2 on the original YOLOv8-pose network,integrates the GhostConv module into the neck to reduce network computation,and addes SEAttention mechanism in the network layer to improve model computation accuracy,and simultaneously uses the WIOU loss function to make the model more effective.Compared with the original model,the improved model achieved a 0.30%improvement in target box detection accuracy,a 33.70%reduction in floating-point computation,and a 38.55%reduction in parameter size,and model size reduced by 37.18%.The experimental results show that under different lightweight networks and feature fusion methods,and under different attention mechanisms,the improved SGS-YOLOv8 pose model and the improved SGS-YOLOv8 pose model can significantly reduce the floating-point computation and parameter count of the model while maintaining high detection accuracy,making it easier to deploy the model in the later stage.

YOLOv8ShuffleNetv2Attention mechanismKey point detection

周玉蝶、张春燕、乔印虎、陈泽伟

展开 >

安徽科技学院,安徽 滁州 233100

安徽云龙粮机有限公司,安徽 阜阳 236500

YOLOv8 ShuffleNetv2 注意力机制 关键点检测

滁州市科技计划(2022)滁州市科技计划(2022)教育部中国高校产学研创新基金(2021)安徽省教育厅自然科学研究重大项目安徽省粮食机械乡村振兴协同技术服务中心项目安徽省自然科学基金面上项目

2022ZN0142022ZN0162021BCE010042022AH040238GXXT-2022-0772308085ME178

2024

10.13954/j.cnki.hdu.2024.02.008

杭州电子科技大学学报
杭州电子科技大学

杭州电子科技大学学报

影响因子：0.277
ISSN：1001-9146
年,卷(期)：2024.44(3)
  • 40