From 538b52bb7fc3266a00aa53e2646693c977a05084 Mon Sep 17 00:00:00 2001 From: hjw <1576345902@qq.com> Date: Fri, 6 Sep 2024 03:42:00 +0000 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20Vision/camera=5Fcoordinate?= =?UTF-8?q?=5Fdete.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Vision/camera_coordinate_dete.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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)