dixisouls commited on
Commit
9526207
·
1 Parent(s): 65c0c50

Added yolo files

Browse files
.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(CONFIG["yolo"]["model"])
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)