AI CUP 2023 - Teaching Computers to Watch Badminton Matches
- Run install_pytorch.sh or install the version you like
- Run install_mm.sh or manually install MMDetection and MMPose
- Run install_tracknetv2.sh or manually install TrackNetv2
- Background Extraction (1_background_extraction.py)
- Background Clustering (2_background_clustering.py)
- Ball Detection (3_ball_processing.py)
- Player Pose Detection (4_pose_detection_*.py)
- Train models for each columns (5_train_*.py)
Predict answers of each columns by each models (6_predict_*.py) step by step
(turned out to be unnecessary)
- Edge detection (Canny): X
- Homography: X
- DIY: O
- DIY - "average_method": X
- DIY - "mode_method": X
- DIY - "average_method_with_masked_players": O
- Blob detection: X
- TrackNetv2: X
- TrackNetv2 + postprocess: O
(Backgound Extraction finished needed)
More content in my report but in Chinese version