智能零售柜商品检测数据集：113类商品与YOLO11跨平台训练指南

简介面向智能零售柜商品检测与通用新零售场景的目标检测数据集共5000张真实监控场景采集的高质量商品图片覆盖罐装饮料、袋装零食等常见零售品类标注113个商品类别。标注采用LabelImg完成精度较高提供VOC、COCO、YOLO三种标准格式标签可直接用于YOLO等主流目标检测算法训练也可作为新零售场景检测任务的数据补充。资源包为单个PDF说明文档5.77MB内含数据集基本情况、标注格式说明、部分缩略图截图及实际获取方式同时附赠YOLO11一键训练脚本支持GPU、CPU及MacM芯片三平台运行并附带博主训练日志供参考。使用该数据集可完成从数据准备、模型训练到结果复现的完整流程当前已有413人学习适合具备一定目标检测基础、希望快速获取高质量零售商品数据并投入训练与项目验证的开发者与研究人员。1. 智能零售柜商品检测数据集5000张图、113类商品与三平台YOLO11脚本能解决的问题做智能零售柜的项目最头疼的往往不是模型选型而是数据。市面上的公开数据集要么是通用场景比如COCO的80类要么是货架场景但类别太少真正贴近零售柜监控视角、带完整标注的商品数据非常难找。这份数据集来自真实零售柜监控场景5000张图113个商品类别覆盖罐装饮料、袋装零食等常见SKU标注用的是labelimg同时给出VOC、COCO、YOLO三种格式的标签拿到手不用再做格式转换。更关键的是它附带一套YOLO11一键训练脚本支持GPU、CPU、Mac M芯片三平台这意味着即使你没有NVIDIA显卡也能在Mac上把模型跑起来。适合谁用正在做零售柜商品检测、新零售场景目标检测的算法工程师和研究生以及在找一份能直接训练、类别覆盖度够高的数据集做实验的从业者。2. 数据集解剖从拍摄场景到113类标注的背后逻辑2.1 真实零售柜监控场景与通用数据集的区别这份数据集最核心的价值在于它的拍摄视角和场景分布。智能零售柜的摄像头通常安装在柜体内部上方以俯视或斜俯视角度拍摄层板上的商品光线条件复杂有柜内LED补光也有环境光干扰玻璃门还会造成反光。这种视角和光照在COCO、VOC这些通用数据集里几乎找不到用通用数据集训练出来的模型部署到零售柜上经常出现检测框偏移、小目标漏检的问题。数据集的商品类别集中在罐装饮料类和袋装零食类这也是零售柜的主流SKU。罐装商品的特征是圆柱体侧面有标签纹理袋装零食则是软包装、形状不规则这两种商品的检测难点完全不同。罐装商品容易在边缘处产生反光导致标签区域过曝袋装零食则因为包装褶皱导致轮廓不稳定。数据集能覆盖到这些差异说明采集时是经过场景设计的不是随手拍的图片堆砌。针对真实场景还有一个容易被忽略的点商品之间的遮挡。零售柜里的商品是紧密排列的层板边缘的商品经常被旁边的商品挡住一部分这在训练时对NMS的阈值设置和anchor的设计都有影响。如果你的项目是从零开始做零售柜检测这份数据能帮你省掉大量数据采集和清洗的时间。2.2 labelimg标注与三种格式的内在一致性数据集采用labelimg标注这是目标检测领域最常用的标注工具之一操作简单支持VOC格式的xml输出。标注质量高不高关键看两点一是框是否贴边二是类别是否准确。从数据集的使用体验看大部分标注框都贴合商品轮廓没有明显的偏移和漏标。三种格式的标签并不是独立标注三遍而是基于原始标注自动转换生成。VOC格式是xml文件每个文件对应一张图片内部用object节点记录类别名和四点坐标。COCO格式是单个json文件所有图片和标注信息统一管理内部用images数组和annotations数组分别存图片信息和标注框信息。YOLO格式是txt文件每行一个目标格式为class_id x_center y_center width height坐标值是相对于图片宽度和高度的归一化值。# VOC格式的xml关键结构示例 annotation filenameimg_0001.jpg/filename size width1920/width height1080/height depth3/depth /size object namecoca_cola_330ml/name bndbox xmin356/xmin ymin482/ymin xmax511/xmax ymax618/ymax /bndbox /object /annotation这段xml表示图片img_0001.jpg中有一个类别为coca_cola_330ml的目标检测框左上角坐标是(356, 482)右下角是(511, 618)。xml格式的优点是可读性强适合人工检查和二次编辑缺点是文件数量多每张图一个xml在数据集规模大时文件管理有些繁琐。# COCO格式json的annotations数组核心字段 { images: [ {id: 1, file_name: img_0001.jpg, width: 1920, height: 1080} ], annotations: [ { id: 1, image_id: 1, category_id: 5, bbox: [356, 482, 155, 136], area: 21080, iscrowd: 0 } ], categories: [ {id: 5, name: coca_cola_330ml} ] }COCO格式的bbox记录的是[x, y, width, height]注意不是[x1, y1, x2, y2]这一点在手工处理json时特别容易搞混。category_id从1开始编号与categories数组中的id对应。COCO格式的重要优势是统一管理、加载方便PyTorch的视觉库和Detectron2都原生支持这种格式。# YOLO格式txt每行一个目标 # class_id x_center y_center width height归一化坐标 5 0.2258 0.5093 0.0807 0.1259 5 0.3812 0.4981 0.0786 0.1315 12 0.6094 0.5231 0.0851 0.1183YOLO格式的坐标是归一化的计算方式为x_center (xmin xmax) / 2 / image_widthwidth (xmax - xmin) / image_width。如果用脚本批量生成需要注意浮点精度通常保留6位小数即可过多位数反而没有意义。2.3 113个类别的分布观察拿到数据后我习惯先统计一下各类别的目标数量分布确认是否有类别严重不平衡的问题。零售柜场景下这种情况很常见某个爆款饮料的样本可能占30%而角落里一些冷门零食只有几十张图。这直接影响模型训练时的loss权重和最终检测精度。常见做法是统计每个类别的框数量然后计算类别频率分布。如果发现长尾明显可以在训练参数里调整cls_loss的权重或者在数据加载时做类别重采样。数据集附带的训练日志里应该能看到mAP的per-class结果重点关注那些AP偏低的类别通常是样本量少或外观相似度高比如不同口味的同品牌饮料的类别。3. 三种格式标签的工程化处理坐标转换与一致性校验3.1 格式转换脚本VOC转YOLO的完整实现数据集提供了三种格式的标签但项目实战中经常需要在自己定义的目录结构之间切换。比如你想用Ultralytics YOLO11训练默认的目录结构是images/train和labels/train分开存放而如果你要跑mmdetection可能就需要COCO格式。虽然数据集已经给了三种格式但理解转换逻辑仍然很重要因为你可能会往数据集里补充自己的图片。VOC转YOLO是最常见的需求。VOC的坐标是绝对像素值YOLO需要的是归一化坐标转换时要注意边界情况目标框理论上不会超出图片边界但标注时手滑可能产生越界转换前最好做一次clip操作。import xml.etree.ElementTree as ET import os def voc_to_yolo(xml_file, class_names, img_width, img_height): tree ET.parse(xml_file) root tree.getroot() yolo_lines [] for obj in root.iter(object): name obj.find(name).text if name not in class_names: continue class_id class_names.index(name) bndbox obj.find(bndbox) xmin float(bndbox.find(xmin).text) ymin float(bndbox.find(ymin).text) xmax float(bndbox.find(xmax).text) ymax float(bndbox.find(ymax).text) # 坐标越界保护避免归一化后出现负数或大于1的值 xmin max(0, min(xmin, img_width - 1)) xmax max(0, min(xmax, img_width - 1)) ymin max(0, min(ymin, img_height - 1)) ymax max(0, min(ymax, img_height - 1)) x_center (xmin xmax) / 2 / img_width y_center (ymin ymax) / 2 / img_height w (xmax - xmin) / img_width h (ymax - ymin) / img_height yolo_lines.append(f{class_id} {x_center:.6f} {y_center:.6f} {w:.6f} {h:.6f}) return yolo_lines这段代码的核心步骤是解析xml定位到每个object节点取出类别名和边界框坐标做越界裁剪后计算归一化的中心点和宽高最后格式化为YOLO要求的字符串。class_names参数是你自己定义的类别列表顺序就是最终训练时的类别id顺序。有一个细节新手容易踩VOC的xmin和xmax可能是字符串也可能是float如果用int()强转再除宽度在图片尺寸不是整数时会有精度损失。稳妥的做法是一律用float()解析最后输出时再统一格式化为6位小数。3.2 COCO格式的校验脚本bbox为负值的排查COCO格式因为只有一个大json出问题时排查范围不容易定位。最常见的问题是bbox坐标出现负值或宽高为0。这种情况通常发生在标注时画框不规范或者转换脚本里坐标计算错误。我一般会写一个快速校验脚本import json def check_coco_annotation(json_path, img_size_map): with open(json_path, r) as f: coco_data json.load(f) errors [] for ann in coco_data[annotations]: image_id ann[image_id] bbox ann[bbox] # [x, y, w, h] x, y, w, h bbox img_w, img_h img_size_map.get(image_id, (None, None)) if img_w is None: errors.append(fannotation {ann[id]}: image_id {image_id} not found) continue # 检查坐标越界和宽高异常 if x 0 or y 0: errors.append(fannotation {ann[id]}: bbox 坐标出现负值: {bbox}) if w 0 or h 0: errors.append(fannotation {ann[id]}: bbox 宽高异常: {bbox}) if x w img_w or y h img_h: errors.append(fannotation {ann[id]}: bbox 超出图片边界: {bbox}) return errors # 使用示例 img_size_map { 1: (1920, 1080), # image_id: (width, height) 2: (1920, 1080), } errors check_coco_annotation(annotations/instances_train.json, img_size_map) if errors: print(f共发现 {len(errors)} 个问题标注) for e in errors[:20]: print(e) else: print(校验通过)这个脚本的核心思路是建立image_id到图片尺寸的映射然后逐条检查bbox是否越界、宽高是否合法。实际使用时img_size_map要从COCO的images数组里动态构建不用手写。校验脚本的价值在于训练前发现问题否则训练到一半因为标注数据异常导致loss不收敛定位问题要花更多时间。3.3 YOLO格式的类别id映射陷阱YOLO格式的txt文件里存的是类别id这个id是整数对应训练配置文件里的类别列表索引。如果数据集提供了classes.txt或data.yaml用它作为唯一真源不要自己猜。常见的问题是自己新建了一个类别顺序不同的yaml结果所有标签的类别全部错位。训练出来模型能跑但检测结果张冠李戴这种错误最难排查。# data.yaml 示例 names: 0: coca_cola_330ml 1: sprite_330ml 2: lays_classic 3: oreo # ... 共113个类别建议拿到数据集后先做的事是打开data.yaml和任意一个txt标签确认txt里的第一个数字对应的类别名和yaml里索引为这个数字的类别名一致。这一步只需要30秒但能避免训练一天后发现类别全部错位的惨剧。4. YOLO11一键训练脚本GPU与Mac M芯片的配置差异4.1 三平台训练脚本的架构设计数据集附带YOLO11训练脚本号称支持GPU、CPU、Mac M芯片三平台。脚本的核心逻辑就是检测当前运行环境自动选择合适的设备配置然后调用YOLO11的训练入口。这个设计解决了初学者最常见的困境GPU显存不够怎么调batch sizeMac上能不能跑YOLO# 训练脚本的命令行入口示例 python train.py --data data.yaml --epochs 100 --batch-size 16 --imgsz 640脚本内部会先检查torch.cuda.is_available()如果为True则用GPU否则检查torch.backends.mps.is_available()判断是否为Apple Silicon两者都没有就退化为CPU训练。这里的核心是batch-size的自动适配GPU显存越小batch size自动调低。4.2 GPU平台训练显存、batch size与混合精度GPU训练时最关键的参数是batch size和混合精度。YOLO11默认使用AMPAutomatic Mixed Precision即FP16和FP32混合训练能在显存占用基本不变的情况下把训练速度提高一倍左右。# GPU训练显存8GB的显卡推荐配置 python train.py --data data.yaml --epochs 100 --batch-size 16 --imgsz 640 --device 0 --amp--device 0指定使用第一张GPU多卡环境用--device 0,1或--device 0,1,2,3--amp开启混合精度脚本默认开启显式写上更明确--imgsz 640是输入图片尺寸零售柜商品检测建议至少640因为罐装饮料上的小字标签需要足够的分辨率才能检测到8GB显存跑YOLO11ssmall版本batch size最多开到16如果显存不足会自动OOM报错这时把batch size降到8即可。不建议用YOLO11x版本在8GB显存上训练参数量太大batch size只能开到4收敛效果反而不好。4.3 Mac M芯片MPS加速的注意事项Mac M芯片跑YOLO11是可用的靠的是PyTorch的MPSMetal Performance Shaders后端。脚本会自动检测MPS是否可用但这里有几个明显的坑# Mac M芯片训练MPS自动启用 python train.py --data data.yaml --epochs 50 --batch-size 8 --imgsz 640 --device mpsMPS后端的不稳定主要在两个方面一是某些算子还没有实现MPS版本运行时会报not implemented错误二是显存管理机制不同于CUDA遇到内存不足时表现不直观。我的经验是MPS训练时batch-size不要超过8否则容易触发内存压力导致训练中断。另外--workers参数在Mac上建议设为0因为macOS的multiprocessing和PyTorch的DataLoader配合经常出问题训练进程会卡住不动。如果MPS确实跑不了脚本会退化为CPU训练。CPU训练的batch size建议2-4epochs可以适当减少因为速度确实慢用这份5000张图的数据跑100个epoch在M芯片的Mac上大概需要十几个小时建议先用50个epoch验证流程能不能跑通。4.4 训练日志解读mAP、loss曲线和训练结果日志里的参考值数据集提供了博主的训练结果日志这个日志的价值在于给你一个预期基准。如果你的训练结果明显低于日志里的mAP说明数据加载或参数设置有误。# 训练日志关键指标示例 Epoch 50/100: 0.9823 Class Images Instances Box(P) R mAP50 mAP50-95 all 5000 28412 0.947 0.935 0.971 0.863 coca_cola 476 1123 0.953 0.941 0.978 0.874 sprite 382 856 0.938 0.927 0.962 0.851看日志的时候重点关注两点一是mAP50是否达到0.9以上二是mAP50-95和mAP50的差距。如果mAP50很高但mAP50-95明显偏低说明检测框的位置精度不够即框虽然框对了位置但和真实框的IoU不高这种情况和标注框的紧致程度、训练时的anchor设置都有关系。5. 避坑指南数据、训练与跨平台脚本的五个常见问题5.1 标签文件名与图片文件名不一致导致训练时没有目标现象训练日志中显示total targets 0loss没有任何下降趋势。原因YOLO格式要求图片和标签文件的名称完全一致只是扩展名不同。比如img_0001.jpg对应的标签文件必须是img_0001.txt。如果从VOC格式转换时没有严格保持文件名一致或者把文件放在错误的子目录如images/train和labels/train不匹配训练时数据加载器找不到对应的标签文件就会把这张图当作无目标图片处理。解决写一个脚本遍历所有训练图片逐一检查对应的txt标签是否存在。import os def check_label_file(image_dir, label_dir): missing [] for img_file in os.listdir(image_dir): if not img_file.endswith((.jpg, .jpeg, .png)): continue stem os.path.splitext(img_file)[0] txt_file os.path.join(label_dir, stem .txt) if not os.path.exists(txt_file): missing.append(img_file) return missing missing check_label_file(images/train, labels/train) print(f缺失标签的图片数量: {len(missing)}) if missing: print(missing[:10])这个脚本的原理很简单遍历图片目录将图片文件的扩展名替换为txt后检查标签目录下是否存在。如果缺失数量较多优先检查数据集解压是否完整或者labelimg标注时是否漏存了部分xml。如果缺失数量极少可以直接删除这些图片避免影响训练。5.2 batch size设置过大导致GPU显存溢出现象训练刚开始即报错CUDA out of memory或者训练到一半随机中断。原因GPU显存是硬性约束batch size乘以单张图片的显存开销必须小于显存总量。当--imgsz设置为640时YOLO11s在FP16混合精度下单张图的显存开销大约为1.5-2GB16的batch size就需要24-32GB显存8GB的卡根本扛不住。解决按显存容量设置batch size8GB显存用812GB用1624GB用32。同时把--cache参数改为--cache ram或者不加避免缓存图片到显存增加额外开销。另外一个有效手段是降低--imgsz到480代价是精度略有下降但对零售柜这种大目标的场景影响有限。5.3 Mac M芯片训练时MPS算子不支持导致报错现象训练脚本检测到MPS可用但运行到某个epoch时报错错误信息包含RuntimeError: Placeholder storage has not been allocated on MPS device或not implemented for MPS。原因PyTorch的MPS后端还在持续完善中YOLO11的数据增强操作和部分损失计算函数可能没有MPS实现导致前向或反向传播时遇到不支持的算子。解决脚本的降级逻辑应该是自动检测到MPS错误后切换到CPU。如果没有这个逻辑手动在环境变量中禁用MPS# 强制使用CPU训练 export PYTORCH_ENABLE_MPS_FALLBACK1 python train.py --data data.yaml --epochs 50 --batch-size 4 --device cpu或者更简单直接把启动参数--device设为cpu。注意MPS不支持时不要反复重试直接切CPU更节省时间训练速度慢一些但稳定。5.4 类别不平衡导致部分商品类别的AP极低现象训练日志显示整体mAP确实在0.9以上但个别类别的AP只有0.5甚至更低比如某些小众零食类别。原因113个类别中长尾分布明显部分冷门SKU在整个数据集中的目标数量可能只有几十个训练样本数量不足以让模型学会判别这些类别。另外部分类别外观非常相似比如不同口味的同品牌饮料罐体颜色接近模型容易混淆。解决最直接的手段是给这些类别增加数据增强的权重或者复制这些样本生成更多训练数据。另一个思路是调整类别损失的权重# data.yaml中增加类别权重配置YOLO11部分版本支持 cls_weights: 0: 1.0 1: 1.0 # 冷门类别权重调高 45: 3.0 46: 3.0调整权重的逻辑是让模型在计算分类损失时给冷门类别更高的惩罚迫使模型更重视这些类别的特征学习。但权重不宜过大否则可能干扰其他类别的正常收敛。5.5 训练日志显示的mAP与博主日志差异过大现象使用完全相同的参数训练自己的结果比博主日志里的结果低了10个百分点以上。原因最常见的原因是数据集划分不一致。博主可能对训练集、验证集做了特定划分而你的脚本自动随机划分后验证集里恰好含有较多困难样本。另一个可能的原因是训练时--seed参数不同导致随机初始化权重路径不同落在不同的局部最优解。解决使用博主提供的划分文件如果没有固定--seed 42再训练一次。数据集划分对最终结果的影响比很多初学者想象的大得多尤其是类别不平衡时。固定种子的意义是让结果有可复现性方便排查问题。6. 验证与进阶用训练结果反推数据集质量让YOLO11跑得更准训练完成后模型的推理精度已经基本定型但还有三个验证维度和两个进阶技巧值得做。第一个验证维度是混淆矩阵。YOLO11训练结束后会在runs/train/exp/confusion_matrix.png生成混淆矩阵从对角线的亮度分布能直观看到哪些类别互相混淆。如果同类商品的不同口味之间出现系统性错分说明特征空间重叠严重需要在数据层面增加这些类别的差异性样本。第二个维度是绘制PR曲线重点关注尾部——那些AP低的类别在PR曲线上通常表现为召回率上不去说明模型对这些类别的特征提取能力不足可以考虑增加这类商品的增强样本比如模拟不同的摆放角度和遮挡程度。第三个维度是直接做一次推理可视化from ultralytics import YOLO # 加载训练好的权重并进行推理验证 model YOLO(runs/train/exp/weights/best.pt) results model.predict( sourcedataset/images/val/, conf0.25, saveTrue, save_txtTrue, imgsz640 )重点关注labels文件夹下生成的txt预测结果与真实标签逐类对比找出系统性的漏检或误检位置。如果漏检集中在层板边缘的商品可以考虑在训练时增加Mosaic增强的概率如果误检集中在玻璃反光区域可以引入随机亮度和对比度扰动来增强模型的鲁棒性。进阶技巧方面第一个值得做的是在现有113类基础上增量训练。零售柜场景的SKU是动态变化的新品上架后不可能重新标注5000张图。正确做法是冻结前80%的层只微调最后的检测头用几百张新品的标注图做增量训练。这种方式既保留了原有类别的识别能力又能快速适配新商品训练时间只需要几十分钟。第二个技巧是知识蒸馏。如果你拥有一张性能好的大模型作为教师模型把它的输出作为软标签来训练一个小模型可以在几乎不掉精度的情况下把模型体积压缩到原来的四分之一这对部署到嵌入式设备很有价值。最后说一个我自己的习惯每次训练结束我都会保存模型在验证集上的失败case按类别归类放到一个文件夹里持续积累。这些失败case在后续迭代中是最有价值的调试素材比盲目调参要高效得多。这个习惯坚持下来已经成了我项目流程的一部分希望也能帮到你。本文还有配套的精品资源点击获取