changed model from XL to M

tue-robotics · Oct 19, 2023 · ca39649 · ca39649
1 parent ff08f4d
commit ca39649
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ed_sensor_integration/scripts/listener.py b/ed_sensor_integration/scripts/listener.py
@@ -45,7 +45,7 @@
 
 class table_segmentor:
     def __init__(self) -> None:
-        model_path = "~/MEGA/developers/Donal/yolov8x-seg.pt"
+        model_path = "~/MEGA/developers/Donal/yolov8m-seg.pt"
         device = "cuda"
         self.model = YOLO(model_path).to(device)
         self.table_class = 60 #table class defined with index 60 (person = 0)