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'])
|