junpan19 / VSR_TGA

code for paper "Video Super-resolution with Temporal Group Attention"
125 stars 13 forks source link

Python 3.6 PyTorch 1.1

Video Super-resolution with Temporal Group Attention (TGA)

The official source code (partially cleaned) for the [Video Super-resolution with Temporal Group Attention] which is accepted by [CVPR-2020].

framework

Train

We utilize 8 Nvidia Tesla V100 GPUs for training.

python main.py

Test

cd code
unzip TGA-without-align-dla.zip

We utilize 1 P100 GPU for testing. Test the trained model with best performance by

python test.py