添加状态分类和液面分割

2025-09-01 14:14:18 +08:00
parent 6e553f6a20
commit ad52ab9125
2379 changed files with 102501 additions and 1465 deletions
--- a/angle_base_seg/test_line_angle_f.py
+++ b/angle_base_seg/test_line_angle_f.py
@ -5,7 +5,7 @@ import os

 # ------------------ 配置 ------------------
 model_path = '../ultralytics_yolo11-main/runs/train/exp4/weights/best.pt'
-img_folder = '/home/hx/yolo/ultralytics_yolo11-main/dataset1/test'  # 你的图片文件夹路径
+img_folder = '/home/hx/yolo/test_image'  # 你的图片文件夹路径
 output_mask_dir = 'output_masks1'
 os.makedirs(output_mask_dir, exist_ok=True)