Official PyTorch Implementation of paper "Video Extrapolation in Space and Time", ECCV 2022.
Stanford University
First, install pytorch and torchvision following the official doc. Then run
pip install -r requirements.txt
Follow the following links to download the original dataset:
To train the model, run the following command:
python train.py configs/kitti/base.yaml -d kitti_city --single_gpu
To compute quantitative results, run the following command:
python compute_scores.py PATH_TO_CHECKPOINT
To generate qualitative results, run
python vest/inference/show_estate.py PATH_TO_CHECKPOINT
This codebase is built on top of Imaginare.