更新最新直接用point_diff_main

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'
+MODEL_PATH = 'pointn.pt'
-IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226'   # 验证集图片目录
+IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test'   # 验证集图片目录
-LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226'          # 标签目录（与图片同名 .txt）
+LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test'          # 标签目录（与图片同名 .txt）
-OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226/output_images'
+OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test/output_images'
 
 
 IMG_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif', '.webp'}

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