Skip to content

Latest commit

 

History

History
30 lines (18 loc) · 850 Bytes

README.md

File metadata and controls

30 lines (18 loc) · 850 Bytes

pythor-tools

Tools to preprocess THÖR dataset used in Context-free Self-Conditioned GAN for Trajectory Forecasting.

Installation

Install miniconda. Then, you can install all packages required by running:

conda env create -f environment.yml

Then, you need to download the raw THÖR dataset from Zenodo repository.

Filtering and splitting

Firstly, set the config file. Then, run:

python -m pythor-tools.run

Secondly, creating training, validation, and test sets:

python -m pythor-tools.split_sets

In the end, you will have one output folder containing the preprocessed files and the respective training, validation, and test sets.