This is written in Python 3.6. The following packages are needed -
numpy
torch
torchvision
cv2
Pillow
- Use this link to download all the pretrained models.
- Put the contents of each folder, named by the model, in
architecture/{model}/pretrained_models/
- Configure
settings, inmain.py, forinputandoutputfolders. - Put all the images and videos to be SR'ed in
settings['input'] - Choose the model to be used
- Run
main.py - Find all the upscaled media in
settings['output']
- Configure
settingsintest_quality.pyfortest_dataset_folderandoutput_dataset_folder - Download this benchmarking dataset and extract its contents in
setting['test_dataset_folder']. - Change other settings if need be.
- Run
test_quality.py
Intel Core i7-7700HQ CPU @ 2.80 GHZ
NVIDIA GeForce GTX 1050 Ti (4GB)
8GB DDR4 RAM