添加状态分类和液面分割

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

@ -5,7 +5,7 @@ import os
# ====================== 用户配置 ======================
MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_pose/weights/best.pt'
IMAGE_PATH = '/home/hx/yolo/test_image/2.png'
IMAGE_PATH = '/output_masks/3.png'
OUTPUT_DIR = '/home/hx/yolo/output_images' # 保存结果的文件夹
os.makedirs(OUTPUT_DIR, exist_ok=True)