diff --git a/Vision/camera_coordinate_dete.py b/Vision/camera_coordinate_dete.py index 87ca44f..e18f189 100644 --- a/Vision/camera_coordinate_dete.py +++ b/Vision/camera_coordinate_dete.py @@ -618,6 +618,7 @@ class Detection: print("RVC X Camera is not opened!") return 0, None, None, None, None + def get_take_photo_position(self, Height_reduce = 30, width_reduce = 30): """ 检测当前拍照点能否检测到料袋 diff --git a/Vision/camera_coordinate_dete_planevison.py b/Vision/camera_coordinate_dete_planevison.py index a574df0..932870b 100644 --- a/Vision/camera_coordinate_dete_planevison.py +++ b/Vision/camera_coordinate_dete_planevison.py @@ -25,6 +25,7 @@ from Vision.tool.utils import uv_to_XY from Vision.tool.utils import fit_plane_vision import os + class Detection_plane_vsion: def __init__(self, use_openvino_model=False, cameraType = 'RVC'): diff --git a/Vision/tool/utils.py b/Vision/tool/utils.py index 2ca22ac..27ff772 100644 --- a/Vision/tool/utils.py +++ b/Vision/tool/utils.py @@ -151,6 +151,7 @@ def remove_nan(pm, y, x): break return point_x, point_y, point_z + def get_disk_space(path='C:'): usage = psutil.disk_usage(path) diff --git a/get_position_test.py b/get_position_test.py index e4b5ef4..5cec46f 100644 --- a/get_position_test.py +++ b/get_position_test.py @@ -104,6 +104,7 @@ def HIK_Camera_test(): img = cv2.resize(frame, (800, 600)) cv2.imshow('img', img) cv2.waitKey(1) + def detectionBagNum_test(): detectionBagNum = DetectionBagNum() video_path = ''.join([os.getcwd(), '/Vision/model/data/2.mp4'])