更新了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

File diff suppressed because it is too large Load Diff

View File

@ -4,7 +4,8 @@ import os
from ultralytics import YOLO 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' # 验证集图片目录 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 LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/point2/train' # 标签目录(与图片同名 .txt
OUTPUT_DIR = './output_images' OUTPUT_DIR = './output_images'

View File

@ -85,7 +85,7 @@ def robndbox_to_yolo_obb(xml_path, output_dir, class_names):
def main(): def main():
# ==================== 配置区 ==================== # ==================== 配置区 ====================
# ✅ 修改以下路径和类别 # ✅ 修改以下路径和类别
XML_DIR = "/home/hx/桌面/ailai_test/train" # 包含 .xml 文件的目录 XML_DIR = "/home/hx/开发/ailai_image_obb/ailai_pc/annotations.xml" # 包含 .xml 文件的目录
OUTPUT_DIR = "yolo_obb_dataset" # 输出目录 OUTPUT_DIR = "yolo_obb_dataset" # 输出目录
CLASS_NAMES = ["ban", "bag"] # 你的类别列表,顺序即 class_id CLASS_NAMES = ["ban", "bag"] # 你的类别列表,顺序即 class_id
# ============================================== # ==============================================

Binary file not shown.

Binary file not shown.