52Hz commited on
Commit
7e79a15
·
1 Parent(s): 714b0bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -9,11 +9,11 @@ os.system(
9
 
10
  def inference(img):
11
  os.system('mkdir test')
12
- basewidth = 256
13
- wpercent = (basewidth / float(img.size[0]))
14
- hsize = int((float(img.size[1]) * float(wpercent)))
15
- img = img.resize((basewidth, hsize), Image.ANTIALIAS)
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'
 
9
 
10
  def inference(img):
11
  os.system('mkdir test')
12
+ #basewidth = 256
13
+ #wpercent = (basewidth / float(img.size[0]))
14
+ #hsize = int((float(img.size[1]) * float(wpercent)))
15
+ #img = img.resize((basewidth, hsize), Image.ANTIALIAS)
16
+ img.save("test/1.jpg", "JPEG")
17
  os.system(
18
  'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
19
  return 'results/1.png'