qaz812345/TrackNetV3

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

해결하는 문제

TrackNetV3는 시각적 간섭과 일시적인 가림(오클루전)으로 인해 샷을 보이지 않는 일반적인 문제를 극복하여 배드민턴 영상에서 샷의 정확한 추적을 가능하게 설계되었습니다.

작동 방식

이 시스템은 두 가지 주요 모듈을 사용합니다:

  1. 궤적 예측: 이 모듈은 추정된 배경을 보조 데이터로 사용하여 시각적 노이즈를 필터링하고 샷을 식별합니다. 또한 복잡한 상황에서 모델의 강건성을 높이기 위해 mixup 데이터 증강을 활용합니다.
  2. 보정: 샷이 가려졌을 때, 시스템은 예측된 궤적을 분석하여 수리 마스크를 생성하고, 인페인팅을 사용하여 누락된 경로를 채워 궤적이 매끄럽고 연속되도록 보장합니다.

대상 사용자

이 프로젝트는 자동 배드민턴 추적 시스템을 개발하는 스포츠 분석 연구자 및 개발자를 대상으로 합니다.

주요 특징

  • 고정도: 샷 궤적 데이터셋에서 YOLOv7 및 TrackNetV2를 능가하는 정확도, 정밀도, F1 스코어를 달성합니다.
  • 배경 추정: 경기나 라운드의 중앙값 이미지를 사용하여 네트워크가 샷과 배경을 구분할 수 있도록 도와줍니다.
  • 궤적 보정: 전용 보정 모듈(InpaintNet)을 통해 오클루전 문제에 특화하여 대응합니다.
  • 에러 분석 도구: 추적 결과의 세부 평가 및 오류 분석을 가능하게 하는 Dash 기반 애플리케이션을 포함합니다.

관련

  • 프로젝트
  • 프로젝트
  • 프로젝트