huangxin/ailai_image_point_diff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
琉璃月光
2026-01-04 13:36:07 +08:00
parent ebcd7dcdcb
commit 68b14a4bec
69 changed files with 13 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
ailai_pc/point_test.py
View File

@ -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'
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251212' # 👈 修改为你的图像文件夹路径
OUTPUT_DIR = './output_images' # 保存结果的文件夹
MODEL_PATH = 'best.pt'
IMAGE_SOURCE_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226' # 👈 修改为你的图像文件夹路径
OUTPUT_DIR = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/20251226/output_images' # 保存结果的文件夹
# 支持的图像扩展名
IMG_EXTENSIONS = {'.png', '.jpg', '.jpeg', '.bmp', '.tiff', '.tif', '.webp'}