From ce061d784060c01536c11860a51a6a26b854ad8f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=90=89=E7=92=83=E6=9C=88=E5=85=89?= <15630071+llyg777@user.noreply.gitee.com> Date: Thu, 8 Jan 2026 17:25:14 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E6=9C=80=E6=96=B0=E7=9B=B4?= =?UTF-8?q?=E6=8E=A5=E7=94=A8point=5Fdiff=5Fmain?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ailai_pc/{best12.pt => best_detect.pt} | Bin ailai_pc/point_error_test.py | 8 ++++---- point_diff_main/calculate_diff3.0.py | 2 +- ...culate_diff(可用设备树版本,测试,不用做推理).py | 0 {point_diff_main => point_save}/calculate_diff.py | 0 .../calculate_diff2.0.py | 0 {point_diff_main => point_save}/point_old.rknn | Bin {point_diff_main => point_save}/point_old1.rknn | Bin {point_diff_main => point_save}/point_old2.rknn | Bin 9 files changed, 5 insertions(+), 5 deletions(-) rename ailai_pc/{best12.pt => best_detect.pt} (100%) rename {point_diff_main => point_save}/caculate_diff(可用设备树版本,测试,不用做推理).py (100%) rename {point_diff_main => point_save}/calculate_diff.py (100%) rename {point_diff_main => point_save}/calculate_diff2.0.py (100%) rename {point_diff_main => point_save}/point_old.rknn (100%) rename {point_diff_main => point_save}/point_old1.rknn (100%) rename {point_diff_main => point_save}/point_old2.rknn (100%) 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