Spaces:
Running
Running
Update app.py
Browse files
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
|
20 |
elif model == 'Denoising (real-world)':
|
21 |
-
os.system('python main_test_SRMNet.py --input_dir test --task
|
22 |
elif model == 'Deblurring (motion-blur)':
|
23 |
-
os.system('python main_test_SRMNet.py --input_dir test --task
|
24 |
elif model == 'Dehazing (dense haze)':
|
25 |
-
os.system('python main_test_SRMNet.py --input_dir test --task
|
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)':
|