更新cls的推理修复

ailai_pc/detet_pc.py

@@ -7,7 +7,7 @@ import cv2
 # 配置参数
 # ======================
 MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai_detect2/weights/best.pt'
-IMG_PATH = '4.jpg'
+IMG_PATH = '3.jpg'
 OUTPUT_PATH = 'output_pt.jpg'
 CONF_THRESH = 0.5
 IOU_THRESH = 0.45

ailai_pc/detet_pc_f.py

@@ -9,13 +9,14 @@ from pathlib import Path
 # ======================
 # 配置参数
 # ======================
-MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai_detect/weights/best.pt'
+MODEL_PATH = "/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai_detect3/weights/best.pt"
-INPUT_FOLDER = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/bag'
+IMAGE_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/delet"
-OUTPUT_FOLDER = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/bag'
+INPUT_FOLDER = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/delet"
+OUTPUT_FOLDER = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/delet"
 
 CONF_BUCKETS = [0.93, 0.95]   # ← ⭐ 自己改这里
 IOU_THRESH = 0.45
-CLASS_NAMES = ['bag']
+CLASS_NAMES = ['bag','bag35']
 
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 IMG_SIZE = 640

ailai_pc/inference_results/bag_center.txt

@@ -1 +0,0 @@
-617.1340942382812 225.45236206054688

ailai_pc/inference_results/ban_center.txt

@@ -1 +0,0 @@
-617.1340942382812 225.45236206054688

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 = 'point.pt'
+MODEL_PATH = 'pointn.pt'
-IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251214'   # 验证集图片目录
+IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test'   # 验证集图片目录
-LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251214'          # 标签目录（与图片同名 .txt）
+LABEL_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/test'          # 标签目录（与图片同名 .txt）
-OUTPUT_DIR = './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'}

ailai_pc/point_test.py

@@ -4,9 +4,9 @@ from ultralytics import YOLO
 import os
 
 # ====================== 用户配置 ======================
-MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai/weights/best.pt'
+MODEL_PATH = 'best.pt'
-IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251212'  # 👈 修改为你的图像文件夹路径
+IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226'  # 👈 修改为你的图像文件夹路径
-OUTPUT_DIR = './output_images'  # 保存结果的文件夹
+OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226/output_images'  # 保存结果的文件夹
 
 # 支持的图像扩展名
 IMG_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif', '.webp'}

cls_main/main_cls.py

@@ -46,7 +46,7 @@ def init_rknn_model(model_path):
         ret = rknn_lite.load_rknn(model_path)
         if ret != 0:
             raise RuntimeError(f'Load model failed: {ret}')
-        ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
+        ret = rknn_lite.init_runtime()
         if ret != 0:
             raise RuntimeError(f'Init runtime failed: {ret}')
         _global_rknn_instance = rknn_lite

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
@@ -173,8 +173,8 @@ def visualize_result(image, keypoints, bbox, fixed_point, offset_info, save_path
 
 # ====================== 主函数 ======================
 
-def calculate_offset_from_image(image_path, visualize=False):
+def calculate_offset_from_image(orig, visualize=False):
-    orig = cv2.imread(image_path)
+
     if orig is None:
         return {'success': False, 'message': f'Failed to load image: {image_path}'}
 
@@ -213,7 +213,8 @@ def calculate_offset_from_image(image_path, visualize=False):
 # ====================== 示例调用 ======================
 if __name__ == "__main__":
     image_path = "11.jpg"
-    result = calculate_offset_from_image(image_path, visualize=False)
+    orig = cv2.imread(image_path)
+    result = calculate_offset_from_image(orig, visualize=False)
     if result['success']:
         print(f"Center point: ({result['cx']:.1f}, {result['cy']:.1f})")
         print(f"Offset: DeltaX={result['dx_mm']:+.2f} mm, DeltaY={result['dy_mm']:+.2f} mm")