diff --git a/README.md b/README.md index d48b2c1..3ffa3ae 100644 --- a/README.md +++ b/README.md @@ -34,6 +34,7 @@ pip install -e . #### Space Data - [Space data in Basketball 🏀](https://github.com/open-starlab/PreProcessing/blob/master/preprocessing/sports/space_data/basketball/README.md) - [Space data in Football/Soccer ⚽](https://github.com/open-starlab/PreProcessing/blob/master/preprocessing/sports/space_data/soccer/README.md) +- [Space data in Ultimate 🥏](https://github.com/open-starlab/PreProcessing/blob/master/preprocessing/sports/space_data/ultimate/README.md) ## RoadMap - [x] Release the package @@ -54,6 +55,7 @@ pip install -e . Calvin Yeung
Calvin Yeung

💻 Kenjiro Ide
Kenjiro Ide

💻 Zheng Chen
Zheng Chen

💻 + Shunsuke Iwashita
Shunsuke Iwashita

💻 Keisuke Fujii
Keisuke Fujii

🧑‍💻 diff --git a/preprocessing/sports/space_data/space_class.py b/preprocessing/sports/space_data/space_class.py index d7ded26..52efa74 100644 --- a/preprocessing/sports/space_data/space_class.py +++ b/preprocessing/sports/space_data/space_class.py @@ -2,7 +2,7 @@ class Space_data: # Modified the sports list to only include fully supported providers basketball_data_provider = ["SportVU_NBA"] soccer_data_provider = ["fifa_wc_2022"] - ultimate_data_provider = ["UltimateTrack", "UFA"] + ultimate_data_provider = ["UltimateTrack", "UFATrack"] def __new__(cls, data_provider, *args, **kwargs): if data_provider in cls.basketball_data_provider: diff --git a/preprocessing/sports/space_data/ultimate/README.md b/preprocessing/sports/space_data/ultimate/README.md new file mode 100644 index 0000000..d029456 --- /dev/null +++ b/preprocessing/sports/space_data/ultimate/README.md @@ -0,0 +1,9 @@ +# Space data in Ultimate 🥏 +[![Documentation Status](https://readthedocs.org/projects/openstarlab/badge/?version=latest)](https://openstarlab.readthedocs.io/en/latest/Pre_Processing/Sports/index.html) +## Introduction +This package offers functions to load and preprocess space data from various sources in Ultimate. +## Supported Data Providers +You can find detailed documentation on supported data providers [here](https://openstarlab.readthedocs.io/en/latest/Pre_Processing/Sports/Space_Eval/Data_Provider/Ultimate/index.html). The supported providers include: + +- UFATrack +- UltimateTrack \ No newline at end of file diff --git a/preprocessing/sports/space_data/ultimate/ultimate_space_class.py b/preprocessing/sports/space_data/ultimate/ultimate_space_class.py index 89f78f7..fcd7a8c 100644 --- a/preprocessing/sports/space_data/ultimate/ultimate_space_class.py +++ b/preprocessing/sports/space_data/ultimate/ultimate_space_class.py @@ -18,7 +18,7 @@ def __init__( self.out_path = out_path if self.data_provider == "UltimateTrack": self.tracking_herz = 15 - elif self.data_provider == "UFA": + elif self.data_provider == "UFATrack": self.tracking_herz = 10 def get_files(self): diff --git a/preprocessing/sports/space_data/ultimate/ultimate_space_preprocessing.py b/preprocessing/sports/space_data/ultimate/ultimate_space_preprocessing.py index 75d1858..8c6a65d 100644 --- a/preprocessing/sports/space_data/ultimate/ultimate_space_preprocessing.py +++ b/preprocessing/sports/space_data/ultimate/ultimate_space_preprocessing.py @@ -251,10 +251,10 @@ def create_events_metrica(df, tracking_herz): def create_tracking_metrica(df, team, tracking_herz): """ - Create the Metrica format DataFrame for team tracking data from UFA data + Create the Metrica format DataFrame for team tracking data from UFATrack data Args: - df (DataFrame): The UFA intermediate DataFrame containing tracking data + df (DataFrame): The UFATrack intermediate DataFrame containing tracking data with columns: frame, class, x, y, id, closest team (str): Team designation ("Home" for offense, "Away" for defense) tracking_herz (int): Frequency of tracking data (frames per second)