To download the datasets:
# Download the ETH/UCY and SDD datasets, then untar it in datasets
import gdown
gdown.download("https://drive.google.com/uc?id=1iLFoExptF_DPT9RL1Q3lZq77_f1mnFG2","data.tar,gz",quiet=False)
To train the conditional diffusion model:
python train.py
To test the diffusion model and generate conditional samples:
python generate_samples.py
To evaluate the diffusion model on ADE/FDE metrics on ETH/UCY
python evaluate_ethucy.py