Spaces:
Runtime error
Runtime error
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')
|