qaz812345/TrackNetV3

Implementation of paper - TrackNetV3: Enhancing ShuttleCock Tracking with Augmentations and Trajectory Rectification

何を解決するか

TrackNetV3 は、視覚的な干渉や一時的な遮蔽(オクルージョン)によりシャトルコックが視界から消えるといった一般的な課題を克服し、バドミントン動画におけるシャトルコックの正確な追跡を実現するように設計されています。

仕組み

このシステムは2つの主要モジュールを使用しています:

  1. 軌道予測:このモジュールは、推定された背景を補助データとして使用してシャトルコックを特定し、視覚的なノイズをフィルタリングします。また、複雑な状況下でのモデルの頑健性を高めるために mixup データ拡張を採用しています。
  2. 補正:シャトルコックが遮蔽された場合、システムは予測された軌道を分析して修復マスクを作成し、その後 inpainting を使用して欠落した経路を埋めることで、滑らかで連続した軌道を確保します。

対象ユーザー

このプロジェクトは、自動バドミントン追跡システムを構築するスポーツ分析の研究者や開発者を対象としています。

特徴

  • 高い精度:シャトルコック軌道データセットにおいて、YOLOv7 および TrackNetV2 を上回る精度、適合率、F1スコアを達成。
  • 背景推定:試合やラリーの中央値画像を使用して、ネットワークがシャトルコックと背景を区別しやすくします。
  • 軌道修復:専用の補正モジュール(InpaintNet)により、遮蔽問題に特化して対処。
  • エラー分析ツール:追跡結果の詳細な評価とエラー分析を可能にする Dash ベースのアプリケーションを備えています。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト