Skip to content

Official implementation of paper - No Bells, Just Whistles: Sports Field Registration by Leveraging Geometric Properties

License

Notifications You must be signed in to change notification settings

mguti97/No-Bells-Just-Whistles

Repository files navigation

No Bells, Just Whistles: Sports Field Registration by Leveraging Geometric Properties

Scholar Conference arXiv

Contents


Abstract

Broadcast sports field registration is traditionally addressed as a homography estimation task, mapping the visible image area to a planar field model, predominantly focusing on the main camera shot. Addressing the shortcomings of previous approaches, we propose a novel calibration pipeline enabling camera calibration using a 3D soccer field model and extending the process to assess the multiple-view nature of broadcast videos. Our approach begins with a keypoint generation pipeline derived from SoccerNet dataset annotations, leveraging the geometric properties of the court. Subsequently, we execute classical camera calibration through DLT algorithm in a minimalist fashion, without further refinement. Through extensive experimentation on real-world soccer broadcast datasets such as SoccerNet-Calibration, WorldCup 2014 and TS-WorldCup, our method demonstrates superior performance in both multiple- and single-view 3D camera calibration while maintaining competitive results in homography estimation compared to state-of-the-art techniques

Overview of our proposed framework. Top: Training data generation pipeline. Beginning with SoccerNet annotations, we utilize field line extraction and ellipse fitting to establish a hierarchical structure for computing each set of keypoints. Bottom: The encoder-decoder networks produce heatmaps for keypoints and extremities of soccer field lines to extract their positions in the image space. The obtained keypoint set is augmented with intersections of lines generated by the second model to ensure a sufficient number of points.

Weights

Download the keypoints and line detection model weights for both single-view and multiple-view configurations. Our models are pre-trained on the SoccerNet distribution and further finetuned on the WC14 and TSWC datasets for enhanced performance.

Single-View

Model Finetuned Link
Keypoints X SV_kp
Lines X SV_lines
Keypoints WC14 SV_FT_WC14_kp
Lines WC14 SV_FT_WC14_lines
Keypoints TSWC SV_FT_TSWC_kp
Lines TSWC SV_FT_TSWC_lines

Multiple-View

Model Finetuned Link
Keypoints X MV_kp
Lines X MV_lines

Inference

On video:

python inference.py --weights_kp "SV_kp" --weights_line "SV_lines" --input_path "examples/iniesta_sample.mp4" --input_type "video" --display

On image:

python inference.py --weights_kp "SV_kp" --weights_line "SV_lines" --input_path "examples/messi_sample.png" --input_type "image" --save_path "examples/messi_results.png"

Experiments

Datasets

SoccerNet-Calibration-V3:

To download the 2022 version, switch the downloader's task to "calibration".

from SoccerNet.Downloader import SoccerNetDownloader
mySoccerNetDownloader = SoccerNetDownloader(LocalDirectory="</nfs/data/soccernet>")
mySoccerNetDownloader.downloadDataTask(task="calibration-2023", split=["train","valid","test"])

See https://github.com/MM4SPA/tvcalib to download the camera-type annotations for SoccerNet-Calibration-2022.

WorldCup 2014

mkdir -p datasets/WC-2014/test && cd datasets/WC-2014/test
# Images and provided homography matrices from test split
wget https://nhoma.github.io/data/soccer_data.tar.gz
tar -zxvf soccer_data.tar.gz

See https://github.com/MM4SPA/tvcalib to download the additional segment annotations in SoccerNet-Calibration format.

TS-WorldCup

Download the TS-WorldCup dataset


Metrics

Segment Reprojection Error

See https://github.com/SoccerNet/sn-calibration for details on the evaluation metric.

IoU, Projection error and Reprojection error:

For the Homography Estimation evaluation, we adopt the approach outlined at https://github.com/ericsujw/KpSFR. Minor modifications of the used script can be seen in model/metrics.py


Evaluation

We provide scripts scripts/ to reproduce the paper's results for the presented approach. Make sure to change the dataset location on the bash scripts, default is set as "datasets/calibration-2023/", and the location of model weights, default is set as weights/MV_kp.

#Multi-view camera parameter estimation for SN-Calib-2023
chmod +x scripts/run_pipeline_sn23.sh
./scripts/run_pipeline_sn23.sh

#Single-view camera parameter estimation for SN-Calib-2022
chmod +x scripts/run_pipeline_sn22.sh
./scripts/run_pipeline_sn22.sh

#WorldCup 2014 as camera parameter estimation task
#Change the dataset location inside "scripts/run_pipeline_sn23.sh" file, i.e. "datasets/WC-2014"
#Change the save directory inside "scripts/run_pipeline_sn23.sh" file to "inference/inference_3D/inference_wc14"
chmod +x scripts/run_pipeline_sn22.sh
./scripts/run_pipeline_sn22.sh

#WorldCup 2014 homography estimation task
chmod +x scripts/run_pipeline_wc14.sh
./scripts/run_pipeline_wc14.sh

#TS-WorldCup homography estimation task
chmod +x scripts/run_pipeline_tswc.sh
./scripts/run_pipeline_tswc.sh

Task results will be printed on screen.


Requirements

conda env create -f NBJWCalib.yml
conda activate NBJWCalib

PIP

python -m venv NBJWCalib
source env/bin/activate
pip install -r requirements.txt

Citation

@InProceedings{Gutierrez-Perez_2024_CVPR,
    author    = {Guti\'errez-P\'erez, Marc and Agudo, Antonio},
    title     = {No Bells Just Whistles: Sports Field Registration by Leveraging Geometric Properties},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
    month     = {June},
    year      = {2024},
    pages     = {3325-3334}
}

About

Official implementation of paper - No Bells, Just Whistles: Sports Field Registration by Leveraging Geometric Properties

Resources

License

Stars

Watchers

Forks

Packages

No packages published