Spaces:
Runtime error
Runtime error
jjourney1125
commited on
Commit
•
4200544
1
Parent(s):
fac140f
Update app.py
Browse files
app.py
CHANGED
@@ -285,9 +285,9 @@ def test(img_lq, model, args, window_size):
|
|
285 |
if __name__ == '__main__':
|
286 |
|
287 |
parser = argparse.ArgumentParser()
|
288 |
-
parser.add_argument('--task', type=str, default='
|
289 |
'gray_dn, color_dn, jpeg_car, color_jpeg_car')
|
290 |
-
parser.add_argument('--scale', type=int, default=
|
291 |
parser.add_argument('--noise', type=int, default=15, help='noise level: 15, 25, 50')
|
292 |
parser.add_argument('--jpeg', type=int, default=40, help='scale factor: 10, 20, 30, 40')
|
293 |
parser.add_argument('--training_patch_size', type=int, default=128, help='patch size used in training Swin2SR. '
|
@@ -295,8 +295,8 @@ if __name__ == '__main__':
|
|
295 |
'Images are NOT tested patch by patch.')
|
296 |
parser.add_argument('--large_model', action='store_true', help='use large model, only provided for real image sr')
|
297 |
parser.add_argument('--model_path', type=str,
|
298 |
-
default='
|
299 |
-
parser.add_argument('--folder_lq', type=str, default=
|
300 |
parser.add_argument('--folder_gt', type=str, default=None, help='input ground-truth test image folder')
|
301 |
parser.add_argument('--tile', type=int, default=None, help='Tile size, None for no tile during testing (testing as a whole)')
|
302 |
parser.add_argument('--tile_overlap', type=int, default=32, help='Overlapping of different tiles')
|
|
|
285 |
if __name__ == '__main__':
|
286 |
|
287 |
parser = argparse.ArgumentParser()
|
288 |
+
parser.add_argument('--task', type=str, default='real_sr', help='classical_sr, lightweight_sr, real_sr, '
|
289 |
'gray_dn, color_dn, jpeg_car, color_jpeg_car')
|
290 |
+
parser.add_argument('--scale', type=int, default=4, help='scale factor: 1, 2, 3, 4, 8') # 1 for dn and jpeg car
|
291 |
parser.add_argument('--noise', type=int, default=15, help='noise level: 15, 25, 50')
|
292 |
parser.add_argument('--jpeg', type=int, default=40, help='scale factor: 10, 20, 30, 40')
|
293 |
parser.add_argument('--training_patch_size', type=int, default=128, help='patch size used in training Swin2SR. '
|
|
|
295 |
'Images are NOT tested patch by patch.')
|
296 |
parser.add_argument('--large_model', action='store_true', help='use large model, only provided for real image sr')
|
297 |
parser.add_argument('--model_path', type=str,
|
298 |
+
default='experiments/pretrained_models/Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.pth')
|
299 |
+
parser.add_argument('--folder_lq', type=str, default="test", help='input low-quality test image folder')
|
300 |
parser.add_argument('--folder_gt', type=str, default=None, help='input ground-truth test image folder')
|
301 |
parser.add_argument('--tile', type=int, default=None, help='Tile size, None for no tile during testing (testing as a whole)')
|
302 |
parser.add_argument('--tile_overlap', type=int, default=32, help='Overlapping of different tiles')
|