Spaces:
Runtime error
Runtime error
Geraldine J
commited on
Commit
·
6d34899
1
Parent(s):
9491e06
Update size
Browse files
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
|