部署文件+resize_cls+resize_seg
This commit is contained in:
@ -4,8 +4,8 @@ import numpy as np
|
||||
import os
|
||||
|
||||
# ================== 配置参数 ==================
|
||||
MODEL_PATH = r"/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_obb2/weights/best.pt"
|
||||
IMAGE_SOURCE_DIR = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/f13" # 图像文件夹路径
|
||||
MODEL_PATH = r"/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_obb4/weights/best.pt"
|
||||
IMAGE_SOURCE_DIR = r"/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/obb2/test" # 图像文件夹路径
|
||||
OUTPUT_DIR = "./inference_results" # 输出结果保存路径
|
||||
IMG_EXTENSIONS = {'.jpg', '.jpeg', '.png', '.bmp', '.tif', '.tiff', '.webp'}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user