更新了4000张新图像的point.rknn

This commit is contained in:
琉璃月光
2025-11-25 21:06:20 +08:00
parent fa80419654
commit 1cd0f9c855
5 changed files with 3 additions and 3468 deletions

View File

@ -4,7 +4,8 @@ import os
from ultralytics import YOLO
# ====================== 用户配置 ======================
MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt'
#MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt'
MODEL_PATH = 'pointn.pt'
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/point2/train' # 验证集图片目录
LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/point2/train' # 标签目录(与图片同名 .txt
OUTPUT_DIR = './output_images'