节点文献

基于YOLOv5s的轻量化乒乓球目标检测算法

Lightweight object detection algorithm for table tennis based on YOLOv5s

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 赵英王琦沙捷郭倩玲

【Author】 ZHAO Ying;WANG Qi;SHA Jie;GUO Qianling;College of Information Science and Technology,Beijing University of Chemical Technology;Laboratory and Asset Management Division,Capital University of Physical Education and Sports;Library,Beijing University of Chemical Technology;

【通讯作者】 郭倩玲;

【机构】 北京化工大学信息科学与技术学院首都体育学院实验室与资产管理处北京化工大学图书馆

【摘要】 针对乒乓球目标检测方法易受环境、光线、速度等多种因素干扰导致精度和实时性不佳的问题,提出了一种基于YOLOv5s框架的轻量化乒乓球目标检测算法——SYOLO5(Shuffle-YOLOv5s)。首先,采用改进的ShuffleNetV2网络单元组合重构YOLOv5s主干网络,提高特征提取速度;其次,在特征融合的过程中引入高效通道注意力(ECA)机制,有效提升模型的检测性能;接着,采用SIoU Loss(S-Intersection over Union)作为定位损失函数提升网络的收敛速度和定位精度;最后,贴合乒乓球小尺寸的特点,采用双尺度目标检测,进一步提高模型推理速度。实验结果表明,所提算法与YOLOv5s相比,参数量和计算量分别减少了80%和60%,精确率提升了1.9个百分点。

【Abstract】 Aiming at the problem that the table tennis object detection method was susceptible to interference from various factors such as environment, light, and speed, resulting in poor accuracy and real-time, a lightweight table tennis object detection algorithm Shuffle-YOLOv5s(SYOLO5) was proposed based on the YOLOv5s framework. Firstly, YOLOv5s backbone network was reconstructed by improved ShuffleNetV2 network unit combination to speed up feature extraction.Secondly, the Efficient Channel Attention(ECA) in the process of feature fusion was introduced to improve the detection performance. Thirdly, the convergence speed and positioning accuracy of the network were improved by using SIoU(SIntersection over Union) Loss as the positioning loss function. Finally, dual-scale object detection method was adopted to further improve the reasoning speed of the model based on the small size characteristic of table tennis. Experimental results show that compared with YOLO5s, SYOLO5 reduces the parameter amount and calculation amount by 80% and 60%respectively, and increases precision by 1. 9 percentage points.

【基金】 国家重点研发计划项目(2018YFB1403903)
  • 【文献出处】 计算机应用 ,Journal of Computer Applications , 编辑部邮箱 ,2023年S1期
  • 【分类号】G846;TP391.41
  • 【下载频次】353
节点文献中: 

本文链接的文献网络图示:

本文的引文网络