xiongyi/AutoControlSystem-git
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新 Vision/camera_coordinate_dete.py

Browse Source
This commit is contained in:
hjw
2024-09-06 03:42:00 +00:00
parent 4ac05c232b
commit 538b52bb7f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
Vision/camera_coordinate_dete.py
View File

@ -17,7 +17,9 @@ from Vision.yolo.yolov8_pt_seg import yolov8_segment
class Detection: class Detection:
def __init__(self, model_path, device): def __init__(self):
model_path = './pt_model/best.pt'
device = 'cpu'
self.camera_rvc = camera() self.camera_rvc = camera()
self.model = yolov8_segment() self.model = yolov8_segment()
self.model.load_model(model_path, device) self.model.load_model(model_path, device)