增加部署代码
This commit is contained in:
@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user