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