添加状态分类和液面分割

This commit is contained in:
琉璃月光
2025-09-01 14:14:18 +08:00
parent 6e553f6a20
commit ad52ab9125
2379 changed files with 102501 additions and 1465 deletions

View File

@ -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'