This is the test version of MFDIN. Other versions of the code will be sorted and released as soon as possible.
- Python 3 (Recommend to use Anaconda)
- PyTorch >= 1.1 (PyTorch1.1 is confirmed to be available)
- NVIDIA GPU + CUDA
- Deformable Convolution. We use Charles Shang's DCNv2 implementation. Please first compile it.
cd ./codes/models/archs/dcn python setup.py build develop
- Python packages:
pip install numpy opencv-python lmdb pyyaml
We provide some test sets: download here
Synthetic test set: ../datasets/yk_test
Real old videos: ../datasets/real-old-videos
-
Reproduce results with PSNR/SSIM
python codes/test_with_GT.py
-
Reproduce the effect of the real old videos
python codes/test_oldvids.py
-
The results will be found at:partially inference results
cd ./results