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