arXiv:2501.06472cs.CVcs.AI2025-01被引 5

用注意力机制提升羽毛球实时追踪精度,速度超130帧每秒。

YO-CSA-T: A Real-time Badminton Tracking System Utilizing YOLO Based on Contextual and Spatial Attention

  • 基于YOLOv8s改进,加入上下文与空间注意力增强特征提取
  • 3D轨迹检测系统达90.43% [email protected],实测速度超130 fps
  • 支持预测补帧与视图补偿,适合机器人对抗场景

羽毛球对决机器人需要高精度、实时的羽毛球3D轨迹。但羽毛球飞行速度快,易受环境干扰(如球网、光照),难以实现快速准确的2D检测。本文提出YO-CSA检测网络,通过在YOLOv8s的主干、颈部和头部引入上下文与空间注意力机制,增强全局与局部特征的提取与融合能力。进一步构建包含检测、预测与补偿的实时3D轨迹追踪系统:利用双目视觉将2D坐标映射至3D空间,基于历史数据预测未来3D位置,并重投影回左右视图以更新2D检测约束;同时设计补偿模块填补缺失帧。在自建数据集上测试,YO-CSA达到90.43% [email protected],优于YOLOv8s与YOLO11s;系统在12个测试序列中保持超过130 fps的运行速度。

原文摘要 · Abstract (English)

The 3D trajectory of a shuttlecock required for a badminton rally robot for human-robot competition demands real-time performance with high accuracy. However, the fast flight speed of the shuttlecock, along with various visual effects, and its tendency to blend with environmental elements, such as court lines and lighting, present challenges for rapid and accurate 2D detection. In this paper, we first propose the YO-CSA detection network, which optimizes and reconfigures the YOLOv8s model's backbone, neck, and head by incorporating contextual and spatial attention mechanisms to enhance model's ability in extracting and integrating both global and local features. Next, we integrate three major subtasks, detection, prediction, and compensation, into a real-time 3D shuttlecock trajectory detection system. Specifically, our system maps the 2D coordinate sequence extracted by YO-CSA into 3D space using stereo vision, then predicts the future 3D coordinates based on historical information, and re-projects them onto the left and right views to update the position constraints for 2D detection. Additionally, our system includes a compensation module to fill in missing intermediate frames, ensuring a more complete trajectory. We conduct extensive experiments on our own dataset to evaluate both YO-CSA's performance and system effectiveness. Experimental results show that YO-CSA achieves a high accuracy of 90.43% [email protected], surpassing both YOLOv8s and YOLO11s. Our system performs excellently, maintaining a speed of over 130 fps across 12 test sequences.

目标检测羽毛球追踪实时系统注意力机制

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。