Spaces:

dixisouls
/

scene-graph-generator

Running

dixisouls commited on Mar 17

Commit

82f6566

1 Parent(s): 9526207

Added min confidence in YOLO detections

Files changed (1) hide show

app/scene_graph_service.py CHANGED Viewed

@@ -475,7 +475,7 @@ def detect_objects_yolo(
     yolo_model = YOLO("app/models/yolov8n.pt")
     # Run inference
-    results = yolo_model(image_path)
     detections = results[0]
     # No detections

     yolo_model = YOLO("app/models/yolov8n.pt")
     # Run inference
+    results = yolo_model(image_path, conf=0.65)
     detections = results[0]
     # No detections