更新 Vision/camera_coordinate_dete.py

2024-09-06 03:42:00 +00:00
parent 4ac05c232b
commit 538b52bb7f
1 changed files with 3 additions and 1 deletions
--- a/Vision/camera_coordinate_dete.py
+++ b/Vision/camera_coordinate_dete.py
@ -17,7 +17,9 @@ from Vision.yolo.yolov8_pt_seg import yolov8_segment

 class Detection:

-    def __init__(self, model_path, device):
+    def __init__(self):
+        model_path = './pt_model/best.pt'
+        device = 'cpu'
        self.camera_rvc = camera()
        self.model = yolov8_segment()
        self.model.load_model(model_path, device)