部署文件+resize_cls+resize_seg

This commit is contained in:
琉璃月光
2025-09-11 20:44:35 +08:00
parent 471c718d42
commit a8d117af36
877 changed files with 1736 additions and 12534 deletions

View File

@ -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'}