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

9
ailai_pc/detet_pc_f.py
View File

@ -9,13 +9,14 @@ from pathlib import Path
# ======================
# 配置参数
# ======================
MODEL_PATH = '/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai_detect/weights/best.pt'
INPUT_FOLDER = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/bag'
OUTPUT_FOLDER = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/bag'
MODEL_PATH = "/home/hx/yolo/ultralytics_yolo11-main/runs/train/exp_ailai_detect3/weights/best.pt"
IMAGE_DIR = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/ailaidete/train/delet"
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