添加状态分类和液面分割
This commit is contained in:
@ -46,7 +46,7 @@ def draw_keypoints_and_angle(image, keypoints, angle):
|
||||
|
||||
# ====================== 用户配置 ======================
|
||||
MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_pose/weights/best.pt' # 替换为你的模型路径
|
||||
IMAGE_PATH = '/home/hx/yolo/test_image/1.png' # 替换为你的测试图像路径
|
||||
IMAGE_PATH = '/home/hx/yolo/output_masks/1.jpg' # 替换为你的测试图像路径
|
||||
CONF_THRESHOLD = 0.25
|
||||
SAVE_RESULT = True
|
||||
SAVE_PATH = '/home/hx/yolo/test_image/result_with_angle.jpg'
|
||||
|
||||
Reference in New Issue
Block a user