Update app.py
Browse files
app.py
CHANGED
@@ -56,12 +56,12 @@ def yolov8_inference(
|
|
56 |
model.overrides["iou"] = iou_threshold
|
57 |
results = model.predict(image, imgsz=image_size, return_outputs=True)
|
58 |
|
59 |
-
|
60 |
for result in results:
|
61 |
# Extract bounding boxes (xyxy format)
|
62 |
-
|
63 |
|
64 |
-
return
|
65 |
|
66 |
|
67 |
inputs = [
|
|
|
56 |
model.overrides["iou"] = iou_threshold
|
57 |
results = model.predict(image, imgsz=image_size, return_outputs=True)
|
58 |
|
59 |
+
boxes1 = []
|
60 |
for result in results:
|
61 |
# Extract bounding boxes (xyxy format)
|
62 |
+
boxes1.append(result.boxes.cpu().numpy())
|
63 |
|
64 |
+
return boxes1
|
65 |
|
66 |
|
67 |
inputs = [
|