PyTorch Implementation of "Robust Temporal Super-Resolution for Dynamic Motion Video"
PyTorch Implementation of "Densely Connected Hierarchical Network for Image Denoising", second place winner of sRGB track and third place winner of Raw-RGB track on NTIRE 2019 Challenge on Real Image Denoising. (code)
PyTorch reimplementation of "Optical Flow Estimation using a Spatial Pyramid Network" by Simon Niklaus. (code)
Python 3.6
PyTorch 1.0.0
MATLAB
TensorFlow
TensorBoard
TensorBoardX
TorchSummary
We used REDS VTSR dataset for training.
To generate training data, use the matlab code generate_train.m
As an example, use the following command to use our training codes
python main_tsr.py --lr 1e-4 --step 2 --cuda True --train_data0 ./train_data0.h5 --train_data1 ./train_data1.h5 --train_label ./train_label.h5 --valid_data0 ./valid_data0.h5 --valid_data1 ./valid_data1.h5 --valid_label ./valid_label.h5 --gpu 0,1 --checkpoint ./checkpoint
If you want to train t=1/4, change 9th line of the main_tsr.py to from model_tsr_14, import model_tsr
If you want to train t=1/2, change 9th line of the main_tsr.py to from model_tsr_12, import model_tsr
If you want to train t=3/4, change 9th line of the main_tsr.py to from model_tsr_34, import model_tsr
Also, you have to give right label for the target time.
There are other options you can choose.
Please refer to the code.
As an example, use the following command to use our test codes
python test_tsr.py --cuda True --model0 ./model0.pth --data ./path/to/data --gpu 0 --result0 ./result0/
There are other options you can choose.
Please refer to the code.
To use our pretrained model, please download here
If you have any question about the code or paper, please feel free to contact kkbbbj@gmail.com