更新了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
|
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'
|
||||||
|
|||||||
@ -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.
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