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

@ -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
# ==============================================