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.

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.

File: pboxes.csv

Bounding boxes of the players, represented as width, height and center coordinates, in the same coordinate space of the referenced image.

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/build.py: download the video files and extracts the images the datasets refer to. The video and image files will be stored in media/

• 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