Spaces:
Running
Running
Update model_yolov5.py
Browse files- model_yolov5.py +1 -0
model_yolov5.py
CHANGED
@@ -16,6 +16,7 @@ def load_model(model_path, img_size=640):
|
|
16 |
def inference(model, image):
|
17 |
"""Run inference on image and return annotated image."""
|
18 |
results = model(image, size=model.img_size)
|
|
|
19 |
annotator = Annotator(np.asarray(image))
|
20 |
for *box, _, cls in reversed(results.pred[0]):
|
21 |
# label = f'{model.names[int(cls)]} {conf:.2f}'
|
|
|
16 |
def inference(model, image):
|
17 |
"""Run inference on image and return annotated image."""
|
18 |
results = model(image, size=model.img_size)
|
19 |
+
print(results)
|
20 |
annotator = Annotator(np.asarray(image))
|
21 |
for *box, _, cls in reversed(results.pred[0]):
|
22 |
# label = f'{model.names[int(cls)]} {conf:.2f}'
|