jjourney1125 commited on
Commit
4200544
1 Parent(s): fac140f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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='color_dn', 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=1, 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,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='model_zoo/swin2sr/Swin2SR_ClassicalSR_X2_64.pth')
299
- parser.add_argument('--folder_lq', type=str, default=None, 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')
 
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')