更新了diff的rknn3568

This commit is contained in:
琉璃月光
2025-11-19 18:11:22 +08:00
parent 267aac1605
commit 029fb8c72a
5 changed files with 77 additions and 6 deletions

View File

@ -4,8 +4,8 @@ from ultralytics import YOLO
import os
# ====================== 用户配置 ======================
MODEL_PATH = 'best.pt'
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/point2/val' # 👈 修改为你的图像文件夹路径
MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai/weights/best.pt'
IMAGE_SOURCE_DIR = '/home/hx/开发/ailai_image_obb/ailai_pc/yolo_obb_dataset/test' # 👈 修改为你的图像文件夹路径
OUTPUT_DIR = './output_images' # 保存结果的文件夹
# 支持的图像扩展名