52Hz commited on
Commit
4edb95b
·
1 Parent(s): cbee64e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -15,14 +15,15 @@ os.system('wget https://github.com/FanChiMao/SRMNet-thesis/releases/download/v0.
15
  def inference(img, model):
16
  os.system('mkdir test')
17
  img.save("test/1.png", "PNG")
 
18
  if model == 'Denoising (gaussian)':
19
- os.system('python main_test_SRMNet.py --input_dir test --task Deblurring_motionblur')
20
  elif model == 'Denoising (real-world)':
21
- os.system('python main_test_SRMNet.py --input_dir test --task Dehaze_realworld')
22
  elif model == 'Deblurring (motion-blur)':
23
- os.system('python main_test_SRMNet.py --input_dir test --task Denoise_gaussian')
24
  elif model == 'Dehazing (dense haze)':
25
- os.system('python main_test_SRMNet.py --input_dir test --task Denoise_realworld')
26
  elif model == 'Deraining (rainstreak)':
27
  os.system('python main_test_SRMNet.py --input_dir test --task Deraining_raindrop')
28
  elif model == 'Deraining (raindrop)':
 
15
  def inference(img, model):
16
  os.system('mkdir test')
17
  img.save("test/1.png", "PNG")
18
+
19
  if model == 'Denoising (gaussian)':
20
+ os.system('python main_test_SRMNet.py --input_dir test --task Denoise_gaussian')
21
  elif model == 'Denoising (real-world)':
22
+ os.system('python main_test_SRMNet.py --input_dir test --task Denoise_realworld')
23
  elif model == 'Deblurring (motion-blur)':
24
+ os.system('python main_test_SRMNet.py --input_dir test --task Deblurring_motionblur')
25
  elif model == 'Dehazing (dense haze)':
26
+ os.system('python main_test_SRMNet.py --input_dir test --task Dehaze_realworld')
27
  elif model == 'Deraining (rainstreak)':
28
  os.system('python main_test_SRMNet.py --input_dir test --task Deraining_raindrop')
29
  elif model == 'Deraining (raindrop)':