更新加入料带目标检测，判断料带到位，以及控制滚筒逻辑

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

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'}