增加部署代码

This commit is contained in:
琉璃月光
2025-09-05 14:29:33 +08:00
parent ad52ab9125
commit 471c718d42
951 changed files with 14072 additions and 264 deletions

View File

@ -6,7 +6,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'
img_folder = '/home/hx/yolo/output_masks'
output_mask_dir = 'output_masks1'
os.makedirs(output_mask_dir, exist_ok=True)