添加状态分类和液面分割
This commit is contained in:
@ -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)
|
||||
|
||||
Reference in New Issue
Block a user