Spaces:
Runtime error
Runtime error
commenting out resizing of megadetector
Browse files
app.py
CHANGED
@@ -68,8 +68,8 @@ def draw_keypoints_on_image(image,
|
|
68 |
# Predict detections with MegaDetector v5a model
|
69 |
def predict_md(im, size=640):
|
70 |
# resize image
|
71 |
-
g = (size / max(im.size)) # gain
|
72 |
-
im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
|
73 |
|
74 |
## detect objects
|
75 |
results = MD_model(im) # inference # vars(results).keys()= dict_keys(['imgs', 'pred', 'names', 'files', 'times', 'xyxy', 'xywh', 'xyxyn', 'xywhn', 'n', 't', 's'])
|
|
|
68 |
# Predict detections with MegaDetector v5a model
|
69 |
def predict_md(im, size=640):
|
70 |
# resize image
|
71 |
+
# g = (size / max(im.size)) # gain
|
72 |
+
# im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
|
73 |
|
74 |
## detect objects
|
75 |
results = MD_model(im) # inference # vars(results).keys()= dict_keys(['imgs', 'pred', 'names', 'files', 'times', 'xyxy', 'xywh', 'xyxyn', 'xywhn', 'n', 't', 's'])
|