Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,11 +9,11 @@ os.system(
|
|
9 |
|
10 |
def inference(img):
|
11 |
os.system('mkdir test')
|
12 |
-
basewidth =
|
13 |
wpercent = (basewidth / float(img.size[0]))
|
14 |
hsize = int((float(img.size[1]) * float(wpercent)))
|
15 |
img = img.resize((basewidth, hsize), Image.BILINEAR)
|
16 |
-
img.save("test/1.
|
17 |
os.system(
|
18 |
'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
|
19 |
return 'results/1.png'
|
|
|
9 |
|
10 |
def inference(img):
|
11 |
os.system('mkdir test')
|
12 |
+
basewidth = 512
|
13 |
wpercent = (basewidth / float(img.size[0]))
|
14 |
hsize = int((float(img.size[1]) * float(wpercent)))
|
15 |
img = img.resize((basewidth, hsize), Image.BILINEAR)
|
16 |
+
img.save("test/1.png", "PNG")
|
17 |
os.system(
|
18 |
'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
|
19 |
return 'results/1.png'
|