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