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
, forinput
andoutput
folders. - 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
settings
intest_quality.py
fortest_dataset_folder
andoutput_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