Event Transformer

This is the source code of Token-based Spatiotemporal Representation of the Events.

Environment

Python ≥ 3.7
CUDA 11.8

Installation

pip install -r requirements.txt
python setup.py develop

Require additional install sparseconvnet, pytorch3d and flow_vis_torch

Datasets preparation

See datasets.md

Train

python basicsr/train.py -opt options/train/[DATASETS]/XXX.yml

Test

python basicsr/test.py -opt options/train/[DATASETS]/XXX.yml --pretrained experiments/[YML_NAME]/models/net_g_latest.pth