简介面向工业传送带异物检测场景的目标检测数据集适用于铁棍、垃圾两类常见异物的识别与定位。素材源自监控视频抽帧图片质量接近真实产线环境可用于训练基于YOLO、Faster R-CNN等经典检测模型也可用于算法验证与教学演示。资源包共108个文件包含105张JPG原图与3个JSON标注文件标注采用COCO格式含类别、边界框等关键信息可直接接入主流深度学习框架使用。压缩包整体仅4.65MB文件规模适中便于快速下载与测试流程搭建。目前已有413人学习或查阅适合计算机视觉初学者熟悉COCO数据格式也适合工业视觉项目开发者进行快速原型验证。整体而言这是一份小而精的实战数据资源能够帮助有需要的人缩短数据采集与标注的时间成本更专注于模型训练与识别效果调优。1. 传送带异物检测数据集coco json标注的产物先别急着丢给yolov8一条传送带每天过几万件物料混进去一根铁棍就能把后端的破碎机打碎混进去一团垃圾就能堵住下料口。很多工厂想上视觉检测第一步不是买相机而是找数据集。这个标题给的东西听起来很完整能识别铁棍、能识别垃圾而且标注用的是coco json格式——看起来是“拿来就能训”的工业级数据。但实际落地时我建议你先别急着把数据丢进训练脚本里。常规做法是先把coco json这个黑匣子打开做一次数据体检确认类别分布、标注质量和图像数量真的能支撑训练再决定要不要转成yolov8能直接吃的格式。适合谁做工业质检的算法工程师、刚接产线视觉项目的开发者以及正在找数据集练手的学生。这篇笔记按“看懂格式 → 做数据体检 → 转格式 → 训练 → 避坑 → 验证”的顺序走每一步都有能直接抄的脚本。2. coco json格式解析与数据体检先把这个标注文件看明白2.1 coco json的五个核心字段每张图、每个框分别存在哪coco json是目标检测里最常见的数据交换格式之一它把整个数据集的标注信息放在一个json文件里。你拿到手的“coco json格式标注”打开后通常能看到下面几个顶层字段字段作用异物检测场景里的对应内容info数据集描述信息版本、日期、作者一般训练时不使用licenses图像许可信息可忽略images图像列表每条包含id、file_name、width、heightannotations标注列表每条记录一个目标的bbox、类别id、segmentation等categories类别定义通常只有“铁棍”“垃圾”两类也可能还有“正常物料”你在写数据读取脚本时真正要关心的是images和annotations两个字段。images里的id是全局唯一的annotations通过image_id和images关联。每条annotation里的bbox是[x, y, width, height]单位是像素坐标系原点在图像左上角。categories里的id可以和annotations里的category_id对上但注意它不一定从0开始转换时要做映射。提示很多新手拿到一个coco json第一件事是打开文件翻几眼。文件大了以后编辑器直接卡死翻也是白翻。正确姿势是写脚本统计不是肉眼读文件。2.2 异物检测数据集的特殊性类别极端不平衡和长条目标铁棍和垃圾在图像里几乎是两个极端。铁棍是典型的长条形目标长宽比可能到10:1甚至更高垃圾或者塑料袋则是形状不规则、大小不一的目标。这直接决定了后期训练时imgsz参数、锚框长宽比的设置方向。如果你只是拿一个通用目标检测的默认配置去训大概率铁棍检测框会把整条棍子拦腰截断或者垃圾小目标直接漏检。另外很多工厂场景里“正常物料”的数量远大于“异物”但标注时往往只标注异物。这意味着模型实际上是在做“只对少数类负责”的检测——类别不平衡是必然的。在进入训练之前你至少要搞清楚每类有多少个标注框每张图平均几个框小目标面积小于32×32像素占多少比例这些数据决定了你后面要不要做类别重加权、要不要切图增强。2.3 用脚本统计coco json的类别数量、bbox大小与长宽比拿到一个coco json数据集我一般会跑下面这个统计脚本先把家底盘清楚import json from collections import Counter # 加载coco json文件 with open(annotations/train.json, r, encodingutf-8) as f: coco json.load(f) # 建立image_id - 文件名的映射 id2file {img[id]: img[file_name] for img in coco[images]} id2cat {cat[id]: cat[name] for cat in coco[categories]} cat_counter Counter() bbox_stats {cat_name: {count: 0, small: 0, medium: 0, large: 0, max_wh_ratio: 0, min_wh_ratio: 1} for cat_name in id2cat.values()} for ann in coco[annotations]: cat_name id2cat[ann[category_id]] cat_counter[cat_name] 1 x, y, w, h ann[bbox] area w * h stats bbox_stats[cat_name] stats[count] 1 stats[max_wh_ratio] max(stats[max_wh_ratio], w / (h 1e-6)) stats[min_wh_ratio] min(stats[min_wh_ratio], w / (h 1e-6)) # coco官方小/中/大目标定义32^2 小32^2~96^2 中96^2 大 if area 32 * 32: stats[small] 1 elif area 96 * 96: stats[medium] 1 else: stats[large] 1 print(类别数量统计, dict(cat_counter)) for cat_name, stats in bbox_stats.items(): print(f类别 {cat_name}: 共{stats[count]}个框, 小目标{stats[small]}, f中目标{stats[medium]}, 大目标{stats[large]}, f宽高比范围 {stats[min_wh_ratio]:.2f} ~ {stats[max_wh_ratio]:.2f})脚本的逻辑很简单先读出images和categories的映射关系然后遍历annotations按类别累加数量同时统计bbox的面积区间和宽高比范围。输出结果能直接回答三个问题类别数量是否均衡、小目标占比多少、长条目标有多长。如果你的铁棍类别里宽高比最大值超过8后面训练时就要考虑长宽比适配或增大输入分辨率。注意coco json里的area字段有时候是空或者填错的我习惯用bbox自己算一遍不信任文件里已有的area值。segmentation字段同理训练目标检测时优先用bbox。3. 把coco json转换成yolo训练格式转换脚本与四个边界坑3.1 为什么yolov8不直接吃coco jsonyolov8的detect训练接口不直接读取coco json文件它要求的标注格式是每张图像对应一个同名txt文件每一行写“class cx cy w h”其中cx、cy是目标中心点坐标w、h是宽度和高度这四个值全部归一化到0~1之间。而coco json里存的是像素坐标的[x, y, width, height]。两者的差别不只是单位不同更重要的是yolo格式没有存图像尺寸信息每张图的标注必须归一化否则同一行标注在不同尺寸的图像上含义完全不同。很多开发工具链比如roboflow、ultralytics的转换工具能一键完成coco转yolo但工业场景里我建议自己写一个转换脚本。原因很简单你能完全控制类别映射和异常处理批量转换时有没有丢框、有没有越界每一步都在日志里能看到。工具越是黑匣子出了问题越难查。3.2 coco转yolo的转换脚本含类别映射与图片尺寸读取下面这个脚本可以直接跑支持coco json里有多余类别时只保留你关心的类别import json import os from pathlib import Path def coco_to_yolo(json_path, output_root): with open(json_path, r, encodingutf-8) as f: coco json.load(f) # 构建image_id - 图像信息的映射 id2img {img[id]: img for img in coco[images]} # 构建category_id - 从0开始的连续类别索引 cat_id_map {} for idx, cat in enumerate(coco[categories]): cat_id_map[cat[id]] idx # 生成图片目录和标注目录 img_dir Path(output_root) / images lbl_dir Path(output_root) / labels img_dir.mkdir(parentsTrue, exist_okTrue) lbl_dir.mkdir(parentsTrue, exist_okTrue) # 按image_id分组所有标注 anns_by_img {} for ann in coco[annotations]: image_id ann[image_id] anns_by_img.setdefault(image_id, []).append(ann) for image_id, img in id2img.items(): # 读取图像宽高用于归一化 w_img, h_img img[width], img[height] if w_img 0 or h_img 0: print(f警告图像宽高为0, image_id{image_id}, 跳过) continue # 复制或软链原图到images目录 src_path Path(img[file_name]) if src_path.exists(): os.system(fcp {src_path} {img_dir / src_path.name}) else: print(f警告原始图像不存在 {src_path}) continue txt_path lbl_dir / (Path(img[file_name]).stem .txt) lines [] for ann in anns_by_img.get(image_id, []): x, y, w, h ann[bbox] # 跳过空框或异常框 if w 0 or h 0: print(f警告异常bbox, image_id{image_id}, bbox{ann[bbox]}) continue cat_id cat_id_map.get(ann[category_id]) if cat_id is None: continue # 中心点坐标归一化 cx (x w / 2) / w_img cy (y h / 2) / h_img nw w / w_img nh h / h_img # 归一化后越界保护保留至少1e-6避免边界为0 cx min(max(cx, 0.0), 1.0) cy min(max(cy, 0.0), 1.0) nw min(max(nw, 0.0), 1.0) nh min(max(nh, 0.0), 1.0) lines.append(f{cat_id} {cx:.6f} {cy:.6f} {nw:.6f} {nh:.6f}) with open(txt_path, w, encodingutf-8) as f: f.write(\n.join(lines)) if __name__ __main__: coco_to_yolo(annotations/train.json, coco_to_yolo_output)脚本里有两个细节值得注意。第一类别映射用了enumerate(coco[categories])这意味着最终的类别索引完全由json里categories列表的顺序决定所以转换前要确认这个顺序是不是你想要的。第二归一化公式是按中心点计算的不是直接除以宽高很多新手会在这一步出错。注意如果你拿到的是标注好的铁棍/垃圾数据集转换后建议抽样打开几张图对比一下。我习惯把txt内容打印出来手动算一下某个bbox像素坐标是否正确。3.3 coco转yolo的四个边界坑从类别id配对失败到图片尺寸读错第一个坑categories里的id不是从0开始的。coco的类别id经常是1、2、3而yolo必须从0开始。有些数据集甚至出现id跳过的情况比如只有id1和id3。直接拿category_id当yolo类别索引等于类别全部错位。上面脚本里的cat_id_map就是为了做这件事不要省。第二个坑图像实际尺寸和json里的width、height不一致。有些数据集在生成时把图像resize过但json没有同步更新。如果你用脚本按json里的尺寸归一化训练时读进yolov8的图尺寸不一样所有的框整体偏移。转换脚本里的第一个检查就是宽高是否为0但更稳妥的做法是在转换前抽样读取真实图片尺寸核对。第三个坑bbox跨越图像边界。铁棍被裁切到图像边缘时标注框可能在图像外或者部分在图像外。直接归一化可能得到cx、cy超出0~1范围的值yolov8训练时会报错或直接丢弃这些框。解决办法要么保留原框但做clip要么把这类样本单独拎出来做边缘扩展。上面脚本里用的是clip方案工业场景里边缘异物恰恰是最危险的我建议后续专门对边缘目标做数据增强。第四个坑中文类名和特殊字符。coco json的categories里如果是中文名称“铁棍”“垃圾”转换后的data.yaml里可以用中文但yolov8在部分系统上对中文类别名的显示有编码问题。我一般会在训练前统一映射成拼音或英文如iron_stick、garbage等模型部署时再映射回中文标签。4. 用yolov8训练自己的异物检测数据集从目录结构到参数调优4.1 异物检测的数据目录结构images和labels一一对应训练前先把数据集按yolov8的约定整理好。常见做法是下面这个目录结构dataset/ ├── train/ │ ├── images/ │ │ ├── img_001.jpg │ │ └── img_002.jpg │ └── labels/ │ ├── img_001.txt │ └── img_002.txt ├── val/ │ ├── images/ │ └── labels/ └── data.yamltrain和val各自独立不能混在一起。有些数据集只给了训练集没有划分验证集你需要自己按比例拆分。拆分时我强烈建议按来源划分——同一个传送带、同一天连续拍的图片不要同时散落到train和val里否则val会包含和train几乎相同的背景指标虚高。data.yaml的内容很简单path: dataset train: train/images val: val/images names: 0: iron_stick 1: garbage4.2 yolov8训练命令与关键参数分辨率、批次和训练轮数数据准备好后用一条命令启动训练yolo detect train \ datadataset/data.yaml \ modelyolov8n.pt \ epochs150 \ imgsz640 \ batch16 \ patience30 \ projectrun_iron_stick \ nameexp1逐个参数说。model那里选了yolov8n.pt是因为异物检测目标数不多通常每张图0~3个异物小模型足够推理速度快部署到工控机上压力小。如果你发现精度不够再往上升级到yolov8s或yolov8m不要一上来就上大模型。imgsz640是输入分辨率。传送带异物场景比较特殊铁棍这种长条形目标在640分辨率下宽度可能只有几个像素。如果出现漏检第一反应就是把imgsz提到960或1280代价是显存占用和推理时间翻倍产线节拍能不能接受要提前评估。epochs150是训练轮数。工业数据集往往只有几百到几千张图150轮足够看到收敛趋势配合patience30做早停连续30轮验证集指标不再提升就停止训练避免无效空跑。batch16根据显卡显存调整12G显存跑640分辨率没问题显存不够就降到8。4.3 类别不平衡的三种处理手段重加权、复制增强与切图异物检测几乎没有类别平衡的数据集。铁棍可能标注了2000个垃圾只有500个。如果不做任何处理模型会倾向于把“看起来不像铁棍的目标”全部判为背景垃圾的召回率会很难看。第一种做法是给loss加类别权重yolov8里可以在data.yaml中用weights字段但传入后生态不太完善。第二种做法是对少样本类别做复制增强把含垃圾的图复制几份配合随机翻转、亮度调整、模糊等变换相当于变相增加该类别的样本占比。第三种做法是切图把大图切成若干小块垃圾目标在小块里占比变大模型更容易学到特征。我一般优先用切图因为它同时解决了小目标问题和类别不平衡问题而且不需要改loss代码。提示铁棍的样本通常不需要做太多旋转增强——工厂里铁棍在传送带上的角度有一定规律过度旋转会让模型学到“不可能出现的角度”对现场真实数据反而不利。垃圾则相反旋转、缩放、形态变化都可以多做一些。5. 异物检测训练避坑损失下降但指标不动的五类排查5.1 现象训练loss降到很低但验证集mAP只有0.3左右原因模型过拟合到训练集的背景纹理验证集里的光照、角度或物料形态稍有变化模型就不会了。这在传送带场景里尤其常见——同一条传送带的图像背景高度相似模型很容易偷懒直接靠背景判断。解决先看训练集和验证集图像是否来自同一时间段。如果是按时间段重新划分数据然后检查loss曲线里val_loss是不是在中间开始回升回升点就是开始过拟合的点把epochs设为那个点附近或者调大增强参数。还有一种常见情况是标签噪声检查coco json里有没有目标类别标错的框这种硬错误会让模型反复震荡。5.2 现象铁棍被检测成两段或者垃圾小目标完全漏检原因铁棍是长条形目标在yolov8的检测头里如果感受野不够覆盖整根棍子模型只能看到局部会把一个目标拆成多个框。垃圾漏检通常是目标太小下采样倍数太高后小目标特征已经丢失。解决长条目标优先把imgsz调大并针对长宽比做特殊增强——例如把图像沿长边方向做水平翻转让模型见过更多不同朝向的棍子。小目标问题用切图方案把原图切成2×2或4×4的子图再训练推理时同样切图后合并结果。注意合并时要处理重叠区域的重复检测否则现场报警会连响好几次。5.3 现象转换后的标注框整体偏移训练时loss始终不收敛原因这是最“低级”但也最容易踩的坑。coco json里的file_name路径和实际图片对不上或者json里记录的图像尺寸和真实尺寸不同。我遇到过数据集作者用opencv读图分辨率正常但json里把宽高写反了导致yolo格式标注的cx、cy整体混乱。解决转换脚本里加一个自检步骤随机挑选三张图打印出json里的宽高和真实读图的宽高做对比。如果发现不一致以实际读图为准重新归一化。标注偏移还可能是EXIF旋转导致的手机或工业相机拍的图如果带了旋转方向信息读图库不同显示的结果就不同转换前先统一把所有图转正成无EXIF的版本。5.4 现象验证集mAP很高但现场实验效果差得离谱原因验证集和训练集有数据泄漏。常见于数据集作者在划分文件时直接按文件名随机切分同一传送带同一天的图片被同时分到train和val。模型在val上看到的是和train几乎一样的图像指标当然好看一上现场就现原形。解决拿到数据集先查一遍文件命名的规律。如果文件名带时间戳或者图像序列号按时间维度划分——前80%的时间段做train后20%做val模拟“模型没见过新画面”的真实场景。如果文件名没有规律按拍摄批次或光线条件分组后再划分。5.5 现象换一条传送带背景后误检暴增正常物料被识别成垃圾原因模型把传送带背景的纹理、反光特征学进去了。训练集里如果只有一条传送带的数据背景太单一模型很可能会偷懒用背景区分异物——背景颜色不同就把正常物料判成垃圾。解决最有效的办法是增加不同背景的负样本——只标注“没有异物”的传送带图像加入训练集。yolov8里没有背景类概念但可以在训练集中放不带任何box的图片它的作用等价于告诉模型“这些图像里任何目标都没有”。另外部署时建议加一道约束检测框的置信度阈值调高到0.5以上并限制目标最小面积别让一个几像素的噪点触发报警。6. 验证与迭代用可视化失误样本决定下一批数据的采集方向训练完不是结束而是新一轮数据工作的开始。我的习惯是训练结束后立刻用训练好的模型跑一遍验证集把预测结果可视化保存到本地目录然后人工翻看失误案例from ultralytics import YOLO import cv2 model YOLO(run_iron_stick/exp1/weights/best.pt) img cv2.imread(dataset/val/images/img_010.jpg) results model.predict(img, conf0.25, saveTrue) # 打印目标框、类别和置信度 for r in results: for box in r.boxes: cls_id int(box.cls[0]) conf float(box.conf[0]) xyxy box.xyxy[0].tolist() print(f类别{cls_id} 置信度{conf:.2f} 坐标{xyxy})翻失误样本时重点看两类漏检标注里有但模型没检出和误检模型检出了但实际是正常物料。前者说明特征学习不足后者说明背景泛化不够。我通常会把失误样本按原因分类统计哪个原因占比最高——如果80%的漏检都发生在图像边缘说明切图策略没覆盖到位下一轮在边缘区域做平移和镜像增强如果60%的误检和某个反光区域有关说明需要采集该光线条件下的负样本。进阶一点的技巧是统计验证集里每类目标的置信度分布。铁棍检出的平均置信度如果是0.85垃圾只有0.55说明垃圾类别的信号明显偏弱。去看垃圾分类的原始标注确认是小目标占比太高还是标注框本身框选不完整、特征被截断。这个统计过程不花几分钟但能直接决定下一轮是加数据、调分辨率还是清理标签噪声。数据集本身不完美没关系所有工业数据集都要往现场数据上靠。我这几年的血泪经验是别信任何数据集的“开箱即用”宣称拿回来后先花半天做数据体检再花半天做转换和验证划分最后训练看结果决定补数据的方向。先前有一回我拿一份现成的异物检测数据直接训mAP很好看上线第一天就被反光误检干翻了后来老老实实回到数据侧做迭代才算救回来。希望你拿到这份coco json标注的数据集后能直接跳过这个坑希望帮到你。本文还有配套的精品资源点击获取