diff --git a/Vision/camera_coordinate_dete.py b/Vision/camera_coordinate_dete.py index 1be2d14..9a566d6 100644 --- 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)