xinntao/ESRGAN
1
1
Fork 0
mirror of https://github.com/xinntao/ESRGAN.git synced 2024-08-17 08:30:32 +03:00
Code Issues Projects Releases Wiki Activity

add command argv

Browse Source
This commit is contained in:
xinntao 2018-09-05 22:25:43 +08:00
parent e3caa577be
commit 164beed5c7
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
test.py
View File

@ -1,3 +1,4 @@
import sys
import os.path
import glob
import cv2
@ -5,13 +6,12 @@ import numpy as np
import torch
import architecture as arch
mode = 'ESRGAN' # ESRGAN or RRDB_PSNR
mode = sys.argv[1] # ESRGAN or RRDB_PSNR
if mode == 'ESRGAN':
model_path = './models/RRDB_ESRGAN_x4.pth'
elif mode == 'RRDB_PSNR':
model_path = './models/RRDB_PSNR_DF2K_x4.pth'
model_path = './models/RRDB_PSNR_x4.pth'
test_img_folder = 'LR/*'