Spaces:
Sleeping
Sleeping
Added yolo files
Browse files- .DS_Store +0 -0
- app/models/yolov8n.pt +3 -0
- app/scene_graph_service.py +1 -1
.DS_Store
ADDED
|
Binary file (6.15 kB). View file
|
|
|
app/models/yolov8n.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f59b3d833e2ff32e194b5bb8e08d211dc7c5bdf144b90d2c8412c47ccfc83b36
|
| 3 |
+
size 6549796
|
app/scene_graph_service.py
CHANGED
|
@@ -472,7 +472,7 @@ def detect_objects_yolo(
|
|
| 472 |
Bounding boxes in format [x_c, y_c, w, h, class_id] (normalized)
|
| 473 |
"""
|
| 474 |
# Load YOLOv8 model - will download if not present
|
| 475 |
-
yolo_model = YOLO(
|
| 476 |
|
| 477 |
# Run inference
|
| 478 |
results = yolo_model(image_path)
|
|
|
|
| 472 |
Bounding boxes in format [x_c, y_c, w, h, class_id] (normalized)
|
| 473 |
"""
|
| 474 |
# Load YOLOv8 model - will download if not present
|
| 475 |
+
yolo_model = YOLO("app/models/yolov8n.pt")
|
| 476 |
|
| 477 |
# Run inference
|
| 478 |
results = yolo_model(image_path)
|