Geraldine J commited on
Commit
6d34899
1 Parent(s): 9491e06

Update size

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -145,6 +145,10 @@ def yolo(size, iou, conf, im):
145
  # gain
146
  g = (int(size) / max(im.size))
147
  # resize
 
 
 
 
148
  im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)
149
 
150
  model.iou = iou
 
145
  # gain
146
  g = (int(size) / max(im.size))
147
  # resize
148
+ if(max(im.size)>900 and g>0.4):
149
+ g=0.3
150
+ if(max(im.size)>2000 and g>0.4):
151
+ g=0.1
152
  im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS)
153
 
154
  model.iou = iou