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

最新推送

Browse Source
This commit is contained in:
琉璃月光
2026-03-10 13:58:21 +08:00
parent 032479f558
commit eb16eeada3
97 changed files with 16865 additions and 670 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
zhuangtai_class_cls/tuili_f_yuantusave—2cls.py
View File

@ -7,12 +7,12 @@ from ultralytics import YOLO
# ---------------------------
# 配置路径（请按需修改）
# ---------------------------
MODEL_PATH = "xialiao.pt" # 你的二分类模型
INPUT_FOLDER = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/ready" # 输入图像文件夹
OUTPUT_ROOT = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/1/ready/result" # 输出根目录
MODEL_PATH = "gaiban.pt" # 你的二分类模型
INPUT_FOLDER = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/camera02" # 输入图像文件夹
OUTPUT_ROOT = "/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/camera02" # 输出根目录
# 类别映射（必须与训练时的 data.yaml 顺序一致）
CLASS_NAMES = {0: "不合格", 1: "合格"}
CLASS_NAMES = {0: "noready", 1: "ready"}
# ---------------------------
# 批量推理函数（移动原图）