diff --git a/ailai_pc/best12.pt b/ailai_pc/best_detect.pt similarity index 100% rename from ailai_pc/best12.pt rename to ailai_pc/best_detect.pt diff --git a/ailai_pc/point_error_test.py b/ailai_pc/point_error_test.py index 5956a23..d52dfb2 100644 --- a/ailai_pc/point_error_test.py +++ b/ailai_pc/point_error_test.py @@ -5,10 +5,10 @@ from ultralytics import YOLO # ====================== 用户配置 ====================== #MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai2/weights/best.pt' -MODEL_PATH = 'best.pt' -IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226' # 验证集图片目录 -LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226' # 标签目录(与图片同名 .txt) -OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226/output_images' +MODEL_PATH = 'pointn.pt' +IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test' # 验证集图片目录 +LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test' # 标签目录(与图片同名 .txt) +OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test/output_images' IMG_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif', '.webp'} diff --git a/point_diff_main/calculate_diff3.0.py b/point_diff_main/calculate_diff3.0.py index 0a3e2ad..e060251 100644 --- a/point_diff_main/calculate_diff3.0.py +++ b/point_diff_main/calculate_diff3.0.py @@ -37,7 +37,7 @@ def init_rknn_once(model_path): print(f"[ERROR] Failed to load RKNN: {ret}") _rknn_instance = None return None - ret = _rknn_instance.init_runtime(core_mask=RKNNLite.NPU_CORE_0) + ret = _rknn_instance.init_runtime() if ret != 0: print(f"[ERROR] Failed to init RKNN runtime: {ret}") _rknn_instance = None diff --git a/point_diff_main/caculate_diff(可用设备树版本,测试,不用做推理).py b/point_save/caculate_diff(可用设备树版本,测试,不用做推理).py similarity index 100% rename from point_diff_main/caculate_diff(可用设备树版本,测试,不用做推理).py rename to point_save/caculate_diff(可用设备树版本,测试,不用做推理).py diff --git a/point_diff_main/calculate_diff.py b/point_save/calculate_diff.py similarity index 100% rename from point_diff_main/calculate_diff.py rename to point_save/calculate_diff.py diff --git a/point_diff_main/calculate_diff2.0.py b/point_save/calculate_diff2.0.py similarity index 100% rename from point_diff_main/calculate_diff2.0.py rename to point_save/calculate_diff2.0.py diff --git a/point_diff_main/point_old.rknn b/point_save/point_old.rknn similarity index 100% rename from point_diff_main/point_old.rknn rename to point_save/point_old.rknn diff --git a/point_diff_main/point_old1.rknn b/point_save/point_old1.rknn similarity index 100% rename from point_diff_main/point_old1.rknn rename to point_save/point_old1.rknn diff --git a/point_diff_main/point_old2.rknn b/point_save/point_old2.rknn similarity index 100% rename from point_diff_main/point_old2.rknn rename to point_save/point_old2.rknn