更新了4000张新图像的point.rknn
This commit is contained in:
File diff suppressed because it is too large
Load Diff
@ -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'
|
||||
|
||||
@ -85,7 +85,7 @@ def robndbox_to_yolo_obb(xml_path, output_dir, class_names):
|
||||
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" # 输出目录
|
||||
CLASS_NAMES = ["ban", "bag"] # 你的类别列表,顺序即 class_id
|
||||
# ==============================================
|
||||
|
||||
Binary file not shown.
BIN
point_diff_main/point_old1.rknn
Normal file
BIN
point_diff_main/point_old1.rknn
Normal file
Binary file not shown.
Reference in New Issue
Block a user