SquashAI Datasets
This repository contains a collection of datasets and helper scripts related to the SquashAI Project.
A summary of the data contained in the datasets is available here.
Dataset Details
T-box Keypoints
File: tboxes.csv
Coordinates of the four corners of the box enclosing the T-area, represented in the same coordinate space of the referenced image.
Player Bounding Boxes
File: pboxes.csv
Bounding boxes of the players, represented as width, height and center coordinates, in the same coordinate space of the referenced image.
Transposed Player Positions and General Attributes
Files: positions.csv
, attributes.csv
, *.json
General information (player names, shirt color, replay and action tags) and coordinates of the player positions on the squash court, represented on a (-3.2, 3.2)x(-4.875, 4.875) coordinate space. The csv files contain information extracted from the json files. The json files were created with the SquashAI Editor.
Scripts
-
scripts/build.py
: download the video files and extracts the images the datasets refer to. The video and image files will be stored inmedia/
-
scripts/json2csv.py
: extracts information from the json datasets into csv files -
scripts/dist.py
: create a tar.gz archive with metadata and raw image files