尧图网页设计 
简介本资源为面向YOLO系列目标检测算法的仓库工人数据集适用于yolov5、yolov8、yolov9、yolov7、yolov10及yolo11等主流框架可直接用于模型训练与验证测试适合从事工业安全监测、仓储场景目标检测的开发者与研究者使用。压缩包共1870个文件包含623张jpg图像、623个txt标签、623个xml标签及1个yaml配置文件整体约30.25MB图像与标注一一对应目录划分清晰。数据集同时提供YOLO格式与VOC格式两种标签YOLO格式以类别索引和归一化中心点、宽高比例记录目标框便于直接接入训练流程VOC格式则以xml文件保存方便与其他检测框架或标注工具对接。yaml文件可用于快速配置数据路径与类别信息减少环境搭建成本。目前已有43人学习下载适合需要快速验证算法效果或开展仓储安全相关实验的读者参考使用。1. 623 张仓库工人图像够不够训一个能用的 YOLO 检测器623 张带标签的仓库工人图像第一反应多半是“这点数据能训出什么”。我一开始也这么想直到在一个真实仓储场景里用不到 700 张图把安全帽、反光衣、人员三类目标做到产线可用的召回率才意识到数据量从来不是唯一变量——标注质量、场景一致性、增强策略和预训练权重的选择比多几千张脏数据管用得多。这个标题里的warehouse-skj9z数据集核心价值不在“623”这个数字而在于它把仓库工人这个垂直场景的边界框标注做齐了省掉了最耗时的清洗和标注环节。它适合两类人一类是想跑通 YOLO 训练全流程但手头没有业务数据的新手另一类是需要快速验证仓库人员检测方案、又不想从零采数据的算法工程师。下面我按自己实际落地的顺序把从拿到压缩包到跑出可用模型的路径拆开讲。2. 先搞清楚 623 张图能训什么仓库工人检测的任务边界与选型2.1 仓库场景里 YOLO 到底在检测什么仓库工人检测听起来简单实际落地时目标类别划分直接决定模型能不能用。常见做法是把任务拆成三类人员整体、安全帽、反光衣。有些团队还会加“是否佩戴安全帽”这种二分类属性但那是检测之后的逻辑不该塞进检测头。623 张图如果只标一个“person”类模型很快收敛但业务上没法判断合规如果标成 person、helmet、vest 三类标注成本翻倍可换来的安全监控价值也翻倍。这个数据集标题写的是“带标签”我一般会先确认标签格式是 YOLO 的 txt 还是 VOC 的 xml前者直接能用后者要转。类别数建议控制在 3 到 5 之间超过 5 类在 623 张图上每类样本会稀到让损失函数震荡。2.2 为什么选 YOLOv8 而不是 v5 或 v11热词里yolov8训练自己的数据集出现频率很高不是没道理。YOLOv8 的 anchor-free 头对小数据集更友好不需要聚类先验框623 张图省掉了调 anchor 的玄学环节。YOLOv5 虽然生态成熟但它的 anchor 机制在样本少时容易把框回归带偏。YOLOv11 更新可社区踩坑记录还少遇到bn崩溃这类问题可参考的排查帖不多。我一般会选 YOLOv8n 或 YOLOv8s 作为基线n 版参数量小623 张图不容易过拟合s 版精度略高但需要更仔细的早停。预训练权重直接用 COCO 上训好的yolov8n.pt仓库工人和 COCO 里的 person 类有重叠迁移效果比从头训好一大截。2.3 数据划分623 张图别按 8:1:1 硬切623 张图按 8:1:1 切验证集只有 62 张测试集 62 张指标波动会大到让你怀疑人生。我一般按 7:2:1 切训练集 436 张验证集 125 张测试集 62 张。如果场景里存在多个摄像头视角或不同光照条件划分时要保证每个子集都覆盖这些变化否则验证集指标虚高上线就翻车。具体做法是先按场景分组再在组内随机抽避免同一段视频的相邻帧同时进训练和验证造成数据泄漏。import os import random import shutil from pathlib import Path # 原始数据目录结构假设images/ 和 labels/ 平铺存放 src_img Path(warehouse/images) src_lbl Path(warehouse/labels) dst_root Path(dataset) # 按 7:2:1 划分 random.seed(42) all_imgs sorted(src_img.glob(*.jpg)) random.shuffle(all_imgs) n len(all_imgs) train_end int(n * 0.7) val_end int(n * 0.9) splits { train: all_imgs[:train_end], val: all_imgs[train_end:val_end], test: all_imgs[val_end:] } for split, imgs in splits.items(): img_dir dst_root / split / images lbl_dir dst_root / split / labels img_dir.mkdir(parentsTrue, exist_okTrue) lbl_dir.mkdir(parentsTrue, exist_okTrue) for img in imgs: shutil.copy(img, img_dir / img.name) lbl src_lbl / (img.stem .txt) if lbl.exists(): shutil.copy(lbl, lbl_dir / lbl.name)这段脚本做三件事固定随机种子保证可复现、按比例切分、把图像和对应标签一起搬到 YOLO 要求的目录结构。random.seed(42)是后悔药不设种子每次切分结果不同调参时根本分不清是模型变了还是数据变了。img.stem取文件名不含扩展名用来匹配同名 txt 标签。如果标签是 xml 格式需要先用转换脚本转成 YOLO 的归一化坐标格式转换时注意边界框不能超出 0 到 1 范围越界框会让训练直接报错。3. 从压缩包到可训练环境配置与数据格式转换3.1 环境配置CUDA、PyTorch 和 ultralytics 的版本咬合yolo环境配置是新手翻车最多的地方。常见坑是 PyTorch 版本和 CUDA 驱动不匹配报错信息还特别隐晦。我一般先用nvidia-smi看驱动支持的最高 CUDA 版本再去 PyTorch 官网找对应命令。比如驱动显示 CUDA 12.1就装torch2.1.0cu121。ultralytics 包用 pip 装最新稳定版即可它会把 numpy、opencv 这些依赖一起拉下来。如果机器没有 GPUCPU 也能训但 623 张图跑 100 轮大概要几个小时建议至少用一张 8G 显存的卡v100 yolo这种配置当然更快但小数据集上 v100 和 3060 的差距没有想象中大。# 查看驱动支持的 CUDA 版本 nvidia-smi # 创建虚拟环境 conda create -n yolo_wh python3.10 -y conda activate yolo_wh # 安装 PyTorch以 CUDA 12.1 为例 pip install torch2.1.0 torchvision0.16.0 --index-url https://download.pytorch.org/whl/cu121 # 安装 ultralytics pip install ultralytics # 验证安装 yolo checksyolo checks会打印环境信息重点看 CUDA 是否可用、版本是否匹配。如果显示 CPU only说明 PyTorch 装成了 CPU 版要卸载重装。虚拟环境用 conda 而不是 venv是因为 conda 对 CUDA 相关的系统库管理更省心少踩libcudart.so找不到的坑。3.2 标签格式转换VOC 转 YOLO 的四个边界坑如果数据集标签是 xml需要转成 YOLO 的 txt 格式。转换逻辑不复杂但边界处理容易出错。YOLO 格式是类别 x_center y_center width height全部归一化到 0 到 1。四个常见坑坐标算出负数、宽高为 0、类别名和索引对不上、图像尺寸读错。下面脚本把这些问题都兜住了。import xml.etree.ElementTree as ET from pathlib import Path from PIL import Image classes [person, helmet, vest] # 按实际类别顺序改 cls_map {name: i for i, name in enumerate(classes)} def convert(xml_path, img_path, out_path): tree ET.parse(xml_path) root tree.getroot() w int(root.find(size/width).text) h int(root.find(size/height).text) # 用实际图像尺寸兜底防止 xml 里尺寸写错 with Image.open(img_path) as im: w, h im.size lines [] for obj in root.findall(object): name obj.find(name).text.strip() if name not in cls_map: continue xmin float(obj.find(bndbox/xmin).text) ymin float(obj.find(bndbox/ymin).text) xmax float(obj.find(bndbox/xmax).text) ymax float(obj.find(bndbox/ymax).text) # 裁剪到图像范围内 xmin max(0, min(xmin, w)) xmax max(0, min(xmax, w)) ymin max(0, min(ymin, h)) ymax max(0, min(ymax, h)) bw xmax - xmin bh ymax - ymin if bw 1 or bh 1: continue # 跳过无效框 xc (xmin xmax) / 2 / w yc (ymin ymax) / 2 / h nw bw / w nh bh / h lines.append(f{cls_map[name]} {xc:.6f} {yc:.6f} {nw:.6f} {nh:.6f}) out_path.write_text(\n.join(lines))关键点在max(0, min(...))这层裁剪标注员手抖把框拖出图像边界是常事不裁剪归一化后会出现大于 1 的值YOLO 训练时直接抛异常。bw 1过滤掉宽高不足 1 像素的框这种框在缩放后基本消失留着只会引入噪声。类别映射用字典而不是列表 index是为了防止类别顺序写错导致人和安全帽标签互换这种错误训练 loss 看着正常但推理结果完全错乱。3.3 写 data.yaml路径和类别名别写错YOLOv8 训练需要一个 yaml 文件描述数据路径和类别。路径建议用绝对路径相对路径在不同工作目录下跑容易找不到文件。类别名顺序必须和标签里的索引严格对应写反了模型学出来的东西就是错的。path: /home/user/dataset train: train/images val: val/images test: test/images names: 0: person 1: helmet 2: vestpath是根目录train等是相对根目录的子路径。如果验证时发现指标异常低但 loss 在降先检查这个 yaml 里的 names 顺序和转换脚本里的 classes 是否一致。这个错误我见过不止一次排查半天最后发现是类别顺序对不上。4. 训练参数怎么设623 张图上的 batch、学习率和增强策略4.1 小数据集训练的五个必调参数YOLOv8 默认参数是给 COCO 这种大数据集设计的623 张图直接套默认值过拟合和欠拟合都可能出现。下面这组参数是我在类似规模数据集上反复试出来的起点。参数建议值理由epochs150小数据集收敛快150 轮足够观察早停batch168G 显存下 640 分辨率能跑再大容易 OOMimgsz640仓库场景目标较大640 够用1280 收益有限lr00.001比默认 0.01 小防止小数据集上损失震荡patience3030 轮验证指标不升就停省时间lr0调小是关键。默认 0.01 在 COCO 上没问题但 623 张图每轮迭代次数少大学习率会让权重更新过猛loss 曲线上下跳。0.001 配合余弦退火收敛更稳。patience设 30 而不是默认 50是因为小数据集上过拟合来得早多跑 20 轮只是浪费电。4.2 数据增强小数据集的救命稻草也是翻车重灾区YOLOv8 默认开启 mosaic、mixup、HSV 增强。623 张图靠增强能把有效样本量撑起来但增强过头会引入不真实样本。仓库场景里mosaic 把四张图拼一起如果拼出来的工人比例失真模型学到的是错误尺度。我一般把mosaic保持默认 1.0但mixup降到 0.1 甚至关掉因为 mixup 在样本少时容易让目标变得模糊。HSV 增强里hsv_h设 0.015、hsv_s设 0.7、hsv_v设 0.4模拟仓库不同灯光条件。翻转增强fliplr设 0.5 没问题但flipud要关掉工人不会倒挂在天花板上。yolo detect train \ modelyolov8n.pt \ data/home/user/dataset/data.yaml \ epochs150 \ batch16 \ imgsz640 \ lr00.001 \ patience30 \ mixup0.1 \ hsv_h0.015 \ hsv_s0.7 \ hsv_v0.4 \ fliplr0.5 \ flipud0.0 \ projectwarehouse_runs \ nameexp1命令里modelyolov8n.pt会自动下载预训练权重如果网络不通可以提前下好放到当前目录。project和name控制输出目录每次实验换个 name方便对比。训练过程中重点看mAP50-95和val/box_loss如果训练 loss 降但验证 loss 升说明过拟合要么加增强要么减 epochs。4.3 训练中 bn 崩溃和 loss 变 nan 怎么排查热词里yolo训练中bn崩溃是个高频问题。现象是训练几轮后 loss 突然变 nan或者报BatchNorm相关错误。原因通常有三个batch 太小导致 bn 统计量不稳、学习率太大、数据里有脏样本。解决顺序是先查数据用脚本扫一遍标签有没有 nan 或越界值再把 batch 调到 16 以上最后把 lr0 降到 0.0005。如果还崩把amp关掉试混合精度在部分显卡上会引入数值不稳定。from pathlib import Path bad [] for lbl in Path(dataset/train/labels).glob(*.txt): for line in lbl.read_text().strip().split(\n): if not line: continue parts line.split() if len(parts) ! 5: bad.append((lbl.name, 字段数不对)) continue vals [float(x) for x in parts[1:]] if any(v 0 or v 1 for v in vals): bad.append((lbl.name, 坐标越界)) if vals[2] 0 or vals[3] 0: bad.append((lbl.name, 宽高非正)) print(f问题标签数: {len(bad)}) for b in bad[:20]: print(b)这个检查脚本在训练前跑一遍能省掉大量排查时间。字段数不对通常是标注工具导出时多了空格或空行坐标越界是转换时没裁剪宽高非正是框画反了。发现问题标签直接删掉对应图像和标签623 张里少几张不影响。5. 避坑与排查仓库工人检测落地时最容易翻车的五件事5.1 验证集指标很高上线后漏检严重现象是训练日志里 mAP50 到 0.9实际视频里工人走动时频繁漏检。原因多半是数据泄漏——同一段视频的相邻帧被分到了训练集和验证集模型记住了背景而不是目标。解决方法是按视频或时间段分组划分确保验证集的场景和训练集不重叠。如果数据集里没有视频来源信息至少按图像文件名前缀分组同前缀的进同一个子集。5.2 安全帽和人员框重叠导致类别混淆现象是模型把戴安全帽的人头识别成安全帽或者把安全帽识别成人。原因是标注时安全帽框和人员框高度重叠YOLO 的标签分配策略在重叠区域会犹豫。解决方法是检查标注规范人员框应该包含全身安全帽框只框帽子区域两者可以有重叠但不要几乎重合。训练时把overlap_mask保持默认同时适当提高box损失权重让模型更关注框的准确性。5.3 推理速度在 CPU 上慢到无法接受现象是 GPU 上训练好好的部署到没有 GPU 的工控机上单帧推理要几百毫秒。原因是用了 yolov8s 或更大模型CPU 推理吃不消。解决方法是导出 ONNX 或 OpenVINO 格式用yolo export命令同时换 yolov8n。如果还慢把输入分辨率从 640 降到 416仓库场景目标大416 通常够用。导出时注意dynamic参数固定 batch 和尺寸能让推理引擎更好地优化。5.4 混淆矩阵总合不唯一热词里yolo混淆矩阵总合不唯一是个经典困惑。现象是混淆矩阵每行加起来不等于验证集样本总数。原因是 YOLO 的混淆矩阵统计的是检测框匹配结果一个图像可能有多个目标也可能有目标没被匹配上。另外置信度阈值和 IoU 阈值会影响匹配数量。解决方法是看混淆矩阵时同时看confusion_matrix.png和val_batch的可视化结果不要只盯数字。如果确实需要严格的总数对齐把conf阈值调到 0.25 以上减少低置信度框的干扰。5.5 增强后的图像标签错位现象是训练时可视化 batch 发现框和目標对不上。原因是自定义增强脚本里图像做了翻转或缩放但标签没同步变换。YOLOv8 内置增强会自动处理标签但如果自己写了 dataloader 或用了外部增强库必须保证图像和标签用同一套变换矩阵。解决方法是优先用 ultralytics 内置增强非要自定义就用 albumentations 并确保 bbox_params 配置正确。6. 把 623 张图的效果再压榨一点验证方法与进阶技巧训练跑完不是终点623 张图的上限还能再挖一挖。我一般会做三件事用测试集跑一次完整评估、导出模型做实际视频推理、针对漏检样本做定向补标。评估时不要只看 mAP还要看每个类别的 AP 和召回率。仓库场景里安全帽的召回率比人员召回率更重要漏检一个没戴安全帽的人可能出大事。如果安全帽 AP 明显低于人员 AP说明安全帽样本太少或标注质量差优先补标安全帽。# 在测试集上评估 yolo detect val \ modelwarehouse_runs/exp1/weights/best.pt \ data/home/user/dataset/data.yaml \ splittest \ conf0.25 \ iou0.5 # 导出 ONNX 用于部署 yolo export \ modelwarehouse_runs/exp1/weights/best.pt \ formatonnx \ imgsz640 \ simplifyTruesplittest确保用的是没参与训练和调参的数据conf0.25是常用的推理阈值iou0.5控制 NMS 的合并程度。导出 ONNX 时simplifyTrue会做图优化去掉冗余算子推理速度能提升百分之十几。导出后用 onnxruntime 跑一遍确认输出和 PyTorch 一致再上部署环境。进阶技巧里我常用的是「困难样本挖掘」。把测试集里漏检和误检的图像挑出来人工看一遍如果是标注问题就修如果是模型确实学不会就补类似场景的图。623 张图补到 800 张左右往往能带来比调参更大的提升。另一个技巧是「多尺度推理」在推理时把图像缩放到不同尺寸分别检测再合并对小目标召回有帮助代价是速度变慢。仓库工人目标通常不小这个技巧优先级不高。最后说个习惯每次实验的配置、命令、指标都记在一个 markdown 文件里包括失败的那些。623 张图的数据集你可能要跑十几轮实验才能找到最优组合没有记录的话两周后根本想不起来哪组参数对应哪个结果。这个习惯帮我省下了大量重复试错的时间。希望帮到你。本文还有配套的精品资源点击获取 基于623张仓库工人图像的YOLOv8检测器训练与调优实战
简介本资源为面向YOLO系列目标检测算法的仓库工人数据集适用于yolov5、yolov8、yolov9、yolov7、yolov10及yolo11等主流框架可直接用于模型训练与验证测试适合从事工业安全监测、仓储场景目标检测的开发者与研究者使用。压缩包共1870个文件包含623张jpg图像、623个txt标签、623个xml标签及1个yaml配置文件整体约30.25MB图像与标注一一对应目录划分清晰。数据集同时提供YOLO格式与VOC格式两种标签YOLO格式以类别索引和归一化中心点、宽高比例记录目标框便于直接接入训练流程VOC格式则以xml文件保存方便与其他检测框架或标注工具对接。yaml文件可用于快速配置数据路径与类别信息减少环境搭建成本。目前已有43人学习下载适合需要快速验证算法效果或开展仓储安全相关实验的读者参考使用。1. 623 张仓库工人图像够不够训一个能用的 YOLO 检测器623 张带标签的仓库工人图像第一反应多半是“这点数据能训出什么”。我一开始也这么想直到在一个真实仓储场景里用不到 700 张图把安全帽、反光衣、人员三类目标做到产线可用的召回率才意识到数据量从来不是唯一变量——标注质量、场景一致性、增强策略和预训练权重的选择比多几千张脏数据管用得多。这个标题里的warehouse-skj9z数据集核心价值不在“623”这个数字而在于它把仓库工人这个垂直场景的边界框标注做齐了省掉了最耗时的清洗和标注环节。它适合两类人一类是想跑通 YOLO 训练全流程但手头没有业务数据的新手另一类是需要快速验证仓库人员检测方案、又不想从零采数据的算法工程师。下面我按自己实际落地的顺序把从拿到压缩包到跑出可用模型的路径拆开讲。2. 先搞清楚 623 张图能训什么仓库工人检测的任务边界与选型2.1 仓库场景里 YOLO 到底在检测什么仓库工人检测听起来简单实际落地时目标类别划分直接决定模型能不能用。常见做法是把任务拆成三类人员整体、安全帽、反光衣。有些团队还会加“是否佩戴安全帽”这种二分类属性但那是检测之后的逻辑不该塞进检测头。623 张图如果只标一个“person”类模型很快收敛但业务上没法判断合规如果标成 person、helmet、vest 三类标注成本翻倍可换来的安全监控价值也翻倍。这个数据集标题写的是“带标签”我一般会先确认标签格式是 YOLO 的 txt 还是 VOC 的 xml前者直接能用后者要转。类别数建议控制在 3 到 5 之间超过 5 类在 623 张图上每类样本会稀到让损失函数震荡。2.2 为什么选 YOLOv8 而不是 v5 或 v11热词里yolov8训练自己的数据集出现频率很高不是没道理。YOLOv8 的 anchor-free 头对小数据集更友好不需要聚类先验框623 张图省掉了调 anchor 的玄学环节。YOLOv5 虽然生态成熟但它的 anchor 机制在样本少时容易把框回归带偏。YOLOv11 更新可社区踩坑记录还少遇到bn崩溃这类问题可参考的排查帖不多。我一般会选 YOLOv8n 或 YOLOv8s 作为基线n 版参数量小623 张图不容易过拟合s 版精度略高但需要更仔细的早停。预训练权重直接用 COCO 上训好的yolov8n.pt仓库工人和 COCO 里的 person 类有重叠迁移效果比从头训好一大截。2.3 数据划分623 张图别按 8:1:1 硬切623 张图按 8:1:1 切验证集只有 62 张测试集 62 张指标波动会大到让你怀疑人生。我一般按 7:2:1 切训练集 436 张验证集 125 张测试集 62 张。如果场景里存在多个摄像头视角或不同光照条件划分时要保证每个子集都覆盖这些变化否则验证集指标虚高上线就翻车。具体做法是先按场景分组再在组内随机抽避免同一段视频的相邻帧同时进训练和验证造成数据泄漏。import os import random import shutil from pathlib import Path # 原始数据目录结构假设images/ 和 labels/ 平铺存放 src_img Path(warehouse/images) src_lbl Path(warehouse/labels) dst_root Path(dataset) # 按 7:2:1 划分 random.seed(42) all_imgs sorted(src_img.glob(*.jpg)) random.shuffle(all_imgs) n len(all_imgs) train_end int(n * 0.7) val_end int(n * 0.9) splits { train: all_imgs[:train_end], val: all_imgs[train_end:val_end], test: all_imgs[val_end:] } for split, imgs in splits.items(): img_dir dst_root / split / images lbl_dir dst_root / split / labels img_dir.mkdir(parentsTrue, exist_okTrue) lbl_dir.mkdir(parentsTrue, exist_okTrue) for img in imgs: shutil.copy(img, img_dir / img.name) lbl src_lbl / (img.stem .txt) if lbl.exists(): shutil.copy(lbl, lbl_dir / lbl.name)这段脚本做三件事固定随机种子保证可复现、按比例切分、把图像和对应标签一起搬到 YOLO 要求的目录结构。random.seed(42)是后悔药不设种子每次切分结果不同调参时根本分不清是模型变了还是数据变了。img.stem取文件名不含扩展名用来匹配同名 txt 标签。如果标签是 xml 格式需要先用转换脚本转成 YOLO 的归一化坐标格式转换时注意边界框不能超出 0 到 1 范围越界框会让训练直接报错。3. 从压缩包到可训练环境配置与数据格式转换3.1 环境配置CUDA、PyTorch 和 ultralytics 的版本咬合yolo环境配置是新手翻车最多的地方。常见坑是 PyTorch 版本和 CUDA 驱动不匹配报错信息还特别隐晦。我一般先用nvidia-smi看驱动支持的最高 CUDA 版本再去 PyTorch 官网找对应命令。比如驱动显示 CUDA 12.1就装torch2.1.0cu121。ultralytics 包用 pip 装最新稳定版即可它会把 numpy、opencv 这些依赖一起拉下来。如果机器没有 GPUCPU 也能训但 623 张图跑 100 轮大概要几个小时建议至少用一张 8G 显存的卡v100 yolo这种配置当然更快但小数据集上 v100 和 3060 的差距没有想象中大。# 查看驱动支持的 CUDA 版本 nvidia-smi # 创建虚拟环境 conda create -n yolo_wh python3.10 -y conda activate yolo_wh # 安装 PyTorch以 CUDA 12.1 为例 pip install torch2.1.0 torchvision0.16.0 --index-url https://download.pytorch.org/whl/cu121 # 安装 ultralytics pip install ultralytics # 验证安装 yolo checksyolo checks会打印环境信息重点看 CUDA 是否可用、版本是否匹配。如果显示 CPU only说明 PyTorch 装成了 CPU 版要卸载重装。虚拟环境用 conda 而不是 venv是因为 conda 对 CUDA 相关的系统库管理更省心少踩libcudart.so找不到的坑。3.2 标签格式转换VOC 转 YOLO 的四个边界坑如果数据集标签是 xml需要转成 YOLO 的 txt 格式。转换逻辑不复杂但边界处理容易出错。YOLO 格式是类别 x_center y_center width height全部归一化到 0 到 1。四个常见坑坐标算出负数、宽高为 0、类别名和索引对不上、图像尺寸读错。下面脚本把这些问题都兜住了。import xml.etree.ElementTree as ET from pathlib import Path from PIL import Image classes [person, helmet, vest] # 按实际类别顺序改 cls_map {name: i for i, name in enumerate(classes)} def convert(xml_path, img_path, out_path): tree ET.parse(xml_path) root tree.getroot() w int(root.find(size/width).text) h int(root.find(size/height).text) # 用实际图像尺寸兜底防止 xml 里尺寸写错 with Image.open(img_path) as im: w, h im.size lines [] for obj in root.findall(object): name obj.find(name).text.strip() if name not in cls_map: continue xmin float(obj.find(bndbox/xmin).text) ymin float(obj.find(bndbox/ymin).text) xmax float(obj.find(bndbox/xmax).text) ymax float(obj.find(bndbox/ymax).text) # 裁剪到图像范围内 xmin max(0, min(xmin, w)) xmax max(0, min(xmax, w)) ymin max(0, min(ymin, h)) ymax max(0, min(ymax, h)) bw xmax - xmin bh ymax - ymin if bw 1 or bh 1: continue # 跳过无效框 xc (xmin xmax) / 2 / w yc (ymin ymax) / 2 / h nw bw / w nh bh / h lines.append(f{cls_map[name]} {xc:.6f} {yc:.6f} {nw:.6f} {nh:.6f}) out_path.write_text(\n.join(lines))关键点在max(0, min(...))这层裁剪标注员手抖把框拖出图像边界是常事不裁剪归一化后会出现大于 1 的值YOLO 训练时直接抛异常。bw 1过滤掉宽高不足 1 像素的框这种框在缩放后基本消失留着只会引入噪声。类别映射用字典而不是列表 index是为了防止类别顺序写错导致人和安全帽标签互换这种错误训练 loss 看着正常但推理结果完全错乱。3.3 写 data.yaml路径和类别名别写错YOLOv8 训练需要一个 yaml 文件描述数据路径和类别。路径建议用绝对路径相对路径在不同工作目录下跑容易找不到文件。类别名顺序必须和标签里的索引严格对应写反了模型学出来的东西就是错的。path: /home/user/dataset train: train/images val: val/images test: test/images names: 0: person 1: helmet 2: vestpath是根目录train等是相对根目录的子路径。如果验证时发现指标异常低但 loss 在降先检查这个 yaml 里的 names 顺序和转换脚本里的 classes 是否一致。这个错误我见过不止一次排查半天最后发现是类别顺序对不上。4. 训练参数怎么设623 张图上的 batch、学习率和增强策略4.1 小数据集训练的五个必调参数YOLOv8 默认参数是给 COCO 这种大数据集设计的623 张图直接套默认值过拟合和欠拟合都可能出现。下面这组参数是我在类似规模数据集上反复试出来的起点。参数建议值理由epochs150小数据集收敛快150 轮足够观察早停batch168G 显存下 640 分辨率能跑再大容易 OOMimgsz640仓库场景目标较大640 够用1280 收益有限lr00.001比默认 0.01 小防止小数据集上损失震荡patience3030 轮验证指标不升就停省时间lr0调小是关键。默认 0.01 在 COCO 上没问题但 623 张图每轮迭代次数少大学习率会让权重更新过猛loss 曲线上下跳。0.001 配合余弦退火收敛更稳。patience设 30 而不是默认 50是因为小数据集上过拟合来得早多跑 20 轮只是浪费电。4.2 数据增强小数据集的救命稻草也是翻车重灾区YOLOv8 默认开启 mosaic、mixup、HSV 增强。623 张图靠增强能把有效样本量撑起来但增强过头会引入不真实样本。仓库场景里mosaic 把四张图拼一起如果拼出来的工人比例失真模型学到的是错误尺度。我一般把mosaic保持默认 1.0但mixup降到 0.1 甚至关掉因为 mixup 在样本少时容易让目标变得模糊。HSV 增强里hsv_h设 0.015、hsv_s设 0.7、hsv_v设 0.4模拟仓库不同灯光条件。翻转增强fliplr设 0.5 没问题但flipud要关掉工人不会倒挂在天花板上。yolo detect train \ modelyolov8n.pt \ data/home/user/dataset/data.yaml \ epochs150 \ batch16 \ imgsz640 \ lr00.001 \ patience30 \ mixup0.1 \ hsv_h0.015 \ hsv_s0.7 \ hsv_v0.4 \ fliplr0.5 \ flipud0.0 \ projectwarehouse_runs \ nameexp1命令里modelyolov8n.pt会自动下载预训练权重如果网络不通可以提前下好放到当前目录。project和name控制输出目录每次实验换个 name方便对比。训练过程中重点看mAP50-95和val/box_loss如果训练 loss 降但验证 loss 升说明过拟合要么加增强要么减 epochs。4.3 训练中 bn 崩溃和 loss 变 nan 怎么排查热词里yolo训练中bn崩溃是个高频问题。现象是训练几轮后 loss 突然变 nan或者报BatchNorm相关错误。原因通常有三个batch 太小导致 bn 统计量不稳、学习率太大、数据里有脏样本。解决顺序是先查数据用脚本扫一遍标签有没有 nan 或越界值再把 batch 调到 16 以上最后把 lr0 降到 0.0005。如果还崩把amp关掉试混合精度在部分显卡上会引入数值不稳定。from pathlib import Path bad [] for lbl in Path(dataset/train/labels).glob(*.txt): for line in lbl.read_text().strip().split(\n): if not line: continue parts line.split() if len(parts) ! 5: bad.append((lbl.name, 字段数不对)) continue vals [float(x) for x in parts[1:]] if any(v 0 or v 1 for v in vals): bad.append((lbl.name, 坐标越界)) if vals[2] 0 or vals[3] 0: bad.append((lbl.name, 宽高非正)) print(f问题标签数: {len(bad)}) for b in bad[:20]: print(b)这个检查脚本在训练前跑一遍能省掉大量排查时间。字段数不对通常是标注工具导出时多了空格或空行坐标越界是转换时没裁剪宽高非正是框画反了。发现问题标签直接删掉对应图像和标签623 张里少几张不影响。5. 避坑与排查仓库工人检测落地时最容易翻车的五件事5.1 验证集指标很高上线后漏检严重现象是训练日志里 mAP50 到 0.9实际视频里工人走动时频繁漏检。原因多半是数据泄漏——同一段视频的相邻帧被分到了训练集和验证集模型记住了背景而不是目标。解决方法是按视频或时间段分组划分确保验证集的场景和训练集不重叠。如果数据集里没有视频来源信息至少按图像文件名前缀分组同前缀的进同一个子集。5.2 安全帽和人员框重叠导致类别混淆现象是模型把戴安全帽的人头识别成安全帽或者把安全帽识别成人。原因是标注时安全帽框和人员框高度重叠YOLO 的标签分配策略在重叠区域会犹豫。解决方法是检查标注规范人员框应该包含全身安全帽框只框帽子区域两者可以有重叠但不要几乎重合。训练时把overlap_mask保持默认同时适当提高box损失权重让模型更关注框的准确性。5.3 推理速度在 CPU 上慢到无法接受现象是 GPU 上训练好好的部署到没有 GPU 的工控机上单帧推理要几百毫秒。原因是用了 yolov8s 或更大模型CPU 推理吃不消。解决方法是导出 ONNX 或 OpenVINO 格式用yolo export命令同时换 yolov8n。如果还慢把输入分辨率从 640 降到 416仓库场景目标大416 通常够用。导出时注意dynamic参数固定 batch 和尺寸能让推理引擎更好地优化。5.4 混淆矩阵总合不唯一热词里yolo混淆矩阵总合不唯一是个经典困惑。现象是混淆矩阵每行加起来不等于验证集样本总数。原因是 YOLO 的混淆矩阵统计的是检测框匹配结果一个图像可能有多个目标也可能有目标没被匹配上。另外置信度阈值和 IoU 阈值会影响匹配数量。解决方法是看混淆矩阵时同时看confusion_matrix.png和val_batch的可视化结果不要只盯数字。如果确实需要严格的总数对齐把conf阈值调到 0.25 以上减少低置信度框的干扰。5.5 增强后的图像标签错位现象是训练时可视化 batch 发现框和目標对不上。原因是自定义增强脚本里图像做了翻转或缩放但标签没同步变换。YOLOv8 内置增强会自动处理标签但如果自己写了 dataloader 或用了外部增强库必须保证图像和标签用同一套变换矩阵。解决方法是优先用 ultralytics 内置增强非要自定义就用 albumentations 并确保 bbox_params 配置正确。6. 把 623 张图的效果再压榨一点验证方法与进阶技巧训练跑完不是终点623 张图的上限还能再挖一挖。我一般会做三件事用测试集跑一次完整评估、导出模型做实际视频推理、针对漏检样本做定向补标。评估时不要只看 mAP还要看每个类别的 AP 和召回率。仓库场景里安全帽的召回率比人员召回率更重要漏检一个没戴安全帽的人可能出大事。如果安全帽 AP 明显低于人员 AP说明安全帽样本太少或标注质量差优先补标安全帽。# 在测试集上评估 yolo detect val \ modelwarehouse_runs/exp1/weights/best.pt \ data/home/user/dataset/data.yaml \ splittest \ conf0.25 \ iou0.5 # 导出 ONNX 用于部署 yolo export \ modelwarehouse_runs/exp1/weights/best.pt \ formatonnx \ imgsz640 \ simplifyTruesplittest确保用的是没参与训练和调参的数据conf0.25是常用的推理阈值iou0.5控制 NMS 的合并程度。导出 ONNX 时simplifyTrue会做图优化去掉冗余算子推理速度能提升百分之十几。导出后用 onnxruntime 跑一遍确认输出和 PyTorch 一致再上部署环境。进阶技巧里我常用的是「困难样本挖掘」。把测试集里漏检和误检的图像挑出来人工看一遍如果是标注问题就修如果是模型确实学不会就补类似场景的图。623 张图补到 800 张左右往往能带来比调参更大的提升。另一个技巧是「多尺度推理」在推理时把图像缩放到不同尺寸分别检测再合并对小目标召回有帮助代价是速度变慢。仓库工人目标通常不小这个技巧优先级不高。最后说个习惯每次实验的配置、命令、指标都记在一个 markdown 文件里包括失败的那些。623 张图的数据集你可能要跑十几轮实验才能找到最优组合没有记录的话两周后根本想不起来哪组参数对应哪个结果。这个习惯帮我省下了大量重复试错的时间。希望帮到你。本文还有配套的精品资源点击获取