尧图网页设计 
简介这份PDF文档面向工业质检领域的技术开发人员与算法工程师围绕YOLOv11目标检测算法系统讲解高精度缺陷检测与实时分类的完整解决方案适合具备一定深度学习基础、希望将目标检测落地到产线质检场景的读者参考。文档共37页为单一PDF文件压缩包约2.04MB支持目录章节跳转与阅读器左侧大纲快速定位查阅体验较为顺畅。内容从工业质检背景与挑战切入依次展开YOLOv11网络结构、训练过程、数据预处理与增强、特征融合与损失函数改进、模型轻量化与硬件加速等关键技术并给出电子芯片、汽车零部件、纺织品三类缺陷检测案例覆盖系统集成部署与未来趋势展望。目前已有73人学习可帮助读者理清从数据标注、模型调优到部署测试的完整链路掌握实时检测与分类的优化思路。1. 从一张机油盖照片说起YOLOv11 工业质检到底在解决什么产线上一个机油盖从注塑机出来表面有没有缩水、划痕、缺料传统做法是靠人眼在强光工位上看。一个人盯 8 小时前两小时准确率能到 95%后面掉到 80% 都算好的而且漏检的件流到总装就是批量返工。YOLOv11 工业质检这套方案要干的事很具体用目标检测把缺陷框出来同时给出类别再让推理速度跟上产线节拍做到实时分类。它适合两类人——一类是手上已经有缺陷图片、想训一个能落地产线模型的算法工程师另一类是做设备集成的需要把检测结果接到 PLC 或分拣机构上。核心矛盾从来不是模型精度不够而是小目标缺陷 实时性 误检率这三者互相拉扯。这篇笔记就按我实际做过的路径把 YOLOv11 从环境配置、数据标注、训练调参到部署推理整条链路拆开讲重点放在小目标优化和实时分类这两个最容易翻车的地方。2. YOLOv11 环境配置与最小可跑通验证别一上来就装全家桶2.1 为什么选 Ultralytics 这套而不是自己搭工业质检场景对框架的要求其实很朴素训练脚本要短、导出格式要多、推理要能塞进 C 或 TensorRT。YOLOv11 在 Ultralytics 仓库里维护model.train()一行就能起训model.export()能直接出 ONNX、TensorRT Engine、OpenVINO这对后面接产线工控机太关键了。自己用 PyTorch 从零搭检测头光是把 NMS 和 letterbox 对齐就要耗掉一周而且导出时各种算子不支持血泪经验就是——能站在成熟仓库上就别造轮子。选型上还有一个现实理由YOLOv11 的网络结构在 neck 部分做了改进对小目标的特征融合比前几代更友好这对轴承缺陷、机油盖划痕这类只占几十个像素的目标很关键。热词里提到的 hcanet、小波缺陷检测本质都是在特征提取阶段做增强但如果你连 baseline 都没跑通加这些模块只会让排查难度翻倍。2.2 环境配置的具体命令与版本约束我一般用 conda 隔离环境Python 锁 3.10CUDA 用 11.8 或 12.1这两个版本和主流 PyTorch 轮子兼容性最好。# 创建独立环境避免和系统里的 torch 打架 conda create -n yolo11 python3.10 -y conda activate yolo11 # 装 PyTorchCUDA 11.8 对应这个 index pip install torch2.1.0 torchvision0.16.0 --index-url https://download.pytorch.org/whl/cu118 # 装 ultralytics不要指定太老的版本 pip install ultralytics # 验证 GPU 是否真的可用 python -c import torch; print(torch.cuda.is_available(), torch.cuda.get_device_name(0))逻辑说明先隔离环境是因为工业项目经常要在一台机器上跑多个版本模型混装必炸。PyTorch 版本和 CUDA 版本必须对齐cu118这个 index 决定了你装的是 GPU 版还是 CPU 版装错了torch.cuda.is_available()会返回 False训练时默默跑在 CPU 上一个 epoch 跑两小时你还以为是数据量大。参数说明torch2.1.0是我验证过和 ultralytics 兼容稳定的版本再新可能遇到算子导出问题。如果你显卡是 40 系CUDA 12.1 也可以把 index 换成cu121。2.3 用官方权重跑一次推理确认链路通在碰自己的数据之前先用预训练权重跑一张图确认从加载模型到出结果整条链路没问题。from ultralytics import YOLO # 加载预训练权重第一次会自动下载 model YOLO(yolo11n.pt) # 对单张图推理saveTrue 会把结果存到 runs/detect/predict results model.predict( sourcetest.jpg, conf0.25, # 置信度阈值工业场景一般先设低一点看召回 iou0.45, # NMS 的 IoU 阈值 saveTrue, device0 # 用第 0 号 GPU ) # 打印每个框的类别和坐标 for box in results[0].boxes: print(box.cls, box.conf, box.xyxy)逻辑说明这一步的价值是排除环境问题。如果这里报错八成是 CUDA 或 torchvision 没装对而不是你的数据有问题。saveTrue会把带框的图存下来肉眼确认框的位置对不对。参数说明conf0.25是通用起点工业质检里如果漏检代价高可以降到 0.1 先看召回iou0.45控制重叠框合并缺陷密集时调低能保留更多框。device0指定 GPU多卡用device[0,1]。3. 数据标注与格式转换缺陷检测的成败八成在这里3.1 工业缺陷数据的三个特殊性工业质检的数据和 COCO 那种自然图像完全不是一回事。第一缺陷样本极度不平衡一批 1000 个件可能只有 20 个有缺陷正负样本比能到 1:50。第二缺陷尺度极小轴承上的点蚀可能就 10×10 像素机油盖的划痕是细长条。第三背景高度一致同一个工位拍的图背景几乎不变模型很容易学到「背景即正常」的捷径换个光照就崩。所以标注策略上我一般要求标注员对每个缺陷框紧贴边缘不要留太多余量因为小目标本身像素就少框大了特征就被背景稀释。类别定义要克制能合并的缺陷类型就合并类别太多会让每个类的样本更少。3.2 从标注文件到 YOLO 格式的转换脚本YOLO 要的是每张图一个 txt每行类别 中心x 中心y 宽 高全部归一化到 0-1。如果你用 LabelImg 标的是 VOC 的 xml需要转。import os import xml.etree.ElementTree as ET def voc_to_yolo(xml_dir, out_dir, classes): # classes 是类别名列表顺序决定类别 id os.makedirs(out_dir, exist_okTrue) for xml_file in os.listdir(xml_dir): if not xml_file.endswith(.xml): continue tree ET.parse(os.path.join(xml_dir, xml_file)) root tree.getroot() size root.find(size) w int(size.find(width).text) h int(size.find(height).text) lines [] for obj in root.iter(object): cls_name obj.find(name).text if cls_name not in classes: continue # 跳过未定义类别避免训练时报错 cls_id classes.index(cls_name) bbox obj.find(bndbox) x1 float(bbox.find(xmin).text) y1 float(bbox.find(ymin).text) x2 float(bbox.find(xmax).text) y2 float(bbox.find(ymax).text) # 归一化并转成中心点宽高 cx (x1 x2) / 2.0 / w cy (y1 y2) / 2.0 / h bw (x2 - x1) / w bh (y2 - y1) / h lines.append(f{cls_id} {cx:.6f} {cy:.6f} {bw:.6f} {bh:.6f}) out_path os.path.join(out_dir, xml_file.replace(.xml, .txt)) with open(out_path, w) as f: f.write(\n.join(lines)) voc_to_yolo(annotations, labels, [scratch, dent, missing])逻辑说明归一化是关键YOLO 不认绝对像素坐标。classes.index把类别名映射成 id顺序必须和后面 data.yaml 里一致否则训练出来的模型类别全错。跳过未定义类别是防止标注员手滑写了个新名字导致训练中断。参数说明cx, cy是框中心点bw, bh是宽高都除以原图宽高做归一化。保留 6 位小数是为了小目标精度小框归一化后数值很小位数不够会丢精度。3.3 data.yaml 的写法与目录结构# data.yaml path: /data/quality_check # 数据集根目录 train: images/train # 训练集图片相对路径 val: images/val # 验证集图片相对路径 nc: 3 # 类别数 names: # 类别名顺序必须和转换脚本一致 0: scratch 1: dent 2: missing目录结构要严格对齐images/train放图labels/train放同名 txtYOLO 会自动去找。很多人翻车是因为图片叫a.jpg标签叫a.txt但放在了别的目录训练时提示找不到标签模型把所有图当负样本训最后啥也检测不出来。提示划分训练验证集时如果同一批次的图背景几乎一样要按批次划分而不是随机划分否则验证集里全是训练集见过的背景指标虚高。4. 训练调参与小目标优化让模型真正看见小缺陷4.1 起训命令与关键参数yolo detect train \ modelyolo11s.pt \ datadata.yaml \ epochs200 \ imgsz640 \ batch16 \ lr00.01 \ lrf0.01 \ patience50 \ augmentTrue \ cacheTrue \ device0逻辑说明modelyolo11s.pt选 s 而不是 n是因为工业缺陷特征弱n 版容量太小容易欠拟合但如果你要部署到边缘设备n 版更合适这是个精度和速度的权衡。patience50是早停50 个 epoch 验证指标不涨就停省时间。参数说明imgsz640是默认输入尺寸小目标多的话可以提到 1280但显存和速度代价大batch16按显存调爆显存就减半lr00.01初始学习率lrf0.01是最终学习率系数cos 衰减到初始的 1%。cacheTrue把图缓存到内存数据量不大时能显著加速。4.2 小目标优化的四个实操手段热词里 yolov11 小目标优化是高频问题我实际用下来有效的就这几招。第一提高输入分辨率。640 提到 1024 或 1280小目标在特征图上的像素变多检测率能明显涨。代价是推理变慢要算清楚产线节拍能不能接受。第二加 P2 检测层。YOLOv11 默认在 P3/P4/P5 三个尺度检测P2 是更高分辨率的浅层特征专门抓小目标。改法是修改模型 yaml在 head 部分加一个 P2 分支。这个改动会让参数量和计算量上升但小目标召回提升明显。第三数据增强针对性调。默认的 mosaic 对小目标有帮助但copy_paste和mixup在缺陷检测里要慎用因为可能把缺陷贴到不合理的位置模型学到错误上下文。我一般把mosaic保留scale范围调窄避免小目标被缩得更小。第四anchor 或损失权重调整。YOLOv11 是 anchor-free 的但可以在损失里给小目标更高权重。这个改动比较深建议先把前三招用满再考虑。4.3 训练过程怎么判断有没有跑偏看runs/detect/train下的results.png重点看三条线box_loss应该稳定下降如果震荡剧烈说明学习率太大mAP50和mAP50-95的差距如果 50 高但 50-95 低说明框的位置不够准小目标定位差cls_loss如果一直不降可能是类别不平衡太严重。还有一个黑匣子式的现象训练集指标很好验证集一塌糊涂这基本是过拟合或者数据泄漏。工业场景里最常见的是同一张图的不同增强版本同时进了训练和验证集指标虚高上线就崩。解决办法是按件号或批次划分数据集别按图片随机分。5. 推理部署与实时分类从模型到产线节拍5.1 导出成部署格式from ultralytics import YOLO model YOLO(runs/detect/train/weights/best.pt) # 导出 ONNX通用性最好 model.export(formatonnx, imgsz640, simplifyTrue, opset12) # 如果有 TensorRT 环境导出 engine 速度最快 model.export(formatengine, imgsz640, halfTrue, device0)逻辑说明ONNX 适合跨平台工控机上用 ONNX Runtime 就能跑TensorRT Engine 是 NVIDIA 显卡上的极致速度但和显卡型号、CUDA 版本绑定换机器要重新导。simplifyTrue会做图优化去掉冗余算子。halfTrue用 FP16速度翻倍精度掉一点点工业场景一般能接受。参数说明opset12兼容性好太新有些推理引擎不支持。imgsz必须和训练时一致否则精度会掉。5.2 实时分类的两种接法标题里的「实时分类」有两层意思。一层是检测框出来之后根据框的类别做分拣决策这个直接在推理结果里读box.cls就行。另一层是整图分类判断这个件是 OK 还是 NG这个可以用检测结果聚合——只要检出任何缺陷框就判 NG。import cv2 from ultralytics import YOLO model YOLO(best.engine) # 用 TensorRT 引擎 cap cv2.VideoCapture(0) # 接工业相机 while True: ret, frame cap.read() if not ret: break results model.predict(frame, conf0.3, verboseFalse) boxes results[0].boxes # 聚合决策有框就是 NG status NG if len(boxes) 0 else OK # 这里可以把 status 通过串口或 PLC 发出去 print(status, len(boxes)) # 保存推理结果图方便追溯 annotated results[0].plot() cv2.imwrite(flog/{int(cap.get(1))}.jpg, annotated)逻辑说明verboseFalse关掉每帧打印否则日志刷屏拖慢速度。results[0].plot()把框画回原图存下来做质量追溯热词里 yolov11 保存推理结果说的就是这个。聚合决策是最简单的实时分类逻辑实际产线可能还要加连续帧确认避免单帧误检触发停机。参数说明conf0.3比训练时略高是为了压误检产线上误检停机比漏检还烦。相机帧率要和推理速度匹配如果推理一帧要 50ms相机就别开 60fps会积压。5.3 速度与精度的平衡表配置输入尺寸精度 mAP50单帧耗时适用场景yolo11n ONNX640基准快边缘设备、节拍快yolo11s TensorRT FP16640略高很快主流工控机yolo11s TensorRT FP161024高中等小目标多、节拍宽松yolo11m TensorRT1280最高慢离线复检这张表是我实际测出来的相对关系具体数字因硬件而异。选型逻辑是先用 s 640 跑通看漏检率不够就提分辨率或换大模型直到满足产线要求为止。6. 避坑与排查那些让模型上线就崩的细节6.1 训练 loss 正常但验证 mAP 为 0现象训练日志里 box_loss 在降但验证集 mAP 一直是 0。原因九成是标签路径或格式不对YOLO 没读到标签把所有框当背景训。解决检查labels/val目录是否存在且和图片同名打开一个 txt 确认是类别 cx cy w h五列且归一化。用yolo detect train时加--verbose看它报的标签数量。6.2 小目标全部漏检现象大缺陷能检出小缺陷一个都框不出来。原因通常是输入分辨率不够小目标在 P3 特征图上只剩一两个像素。解决先把imgsz提到 1024 试如果还不行就加 P2 检测层。另外检查标注框是不是画得太大把背景也框进去了导致特征被稀释。6.3 换一批产品就误检爆炸现象训练集上好好的换了个批次的件背景稍微变了一点误检率飙升。原因是模型学到了背景捷径没学到缺陷本质。解决训练时加更强的颜色抖动和亮度扰动数据里混入不同批次的正常件做负样本让模型知道背景变化不是缺陷。6.4 导出 ONNX 后精度掉一大截现象PyTorch 里推理正常导出 ONNX 后框的位置偏了或类别错了。原因多半是预处理没对齐YOLO 的 letterbox 填充方式在导出后可能和训练时不一致。解决导出时加simplifyTrue推理时用 ultralytics 自己的predict接口而不是自己写预处理或者严格复现 letterbox 的缩放和填充逻辑。6.5 推理速度达不到产线节拍现象单帧耗时 80ms产线要求 30ms。原因可能是没用 TensorRT、没开 FP16、或者后处理 NMS 在 CPU 上跑。解决导出 engine 并开halfTrue把 NMS 放到 GPU 上或者降低输入分辨率。如果还不行考虑模型剪枝或换 n 版。7. 一个提精度的小技巧用检测结果反哺难例挖掘模型训完第一版之后别急着上线。拿它去跑一批没标注的产线图把置信度在 0.1 到 0.4 之间的框导出来这些就是模型「犹豫」的难例。人工复核这批图把漏标的补上、误检的标成负样本再混进训练集重新训一轮。这个循环做两到三轮mAP 通常能再涨几个点比盲目加数据有效得多。from ultralytics import YOLO model YOLO(best.pt) results model.predict(unlabeled_images, conf0.1, saveFalse) # 收集低置信度样本的文件名 hard_cases [] for r in results: for box in r.boxes: if 0.1 box.conf 0.4: hard_cases.append(r.path) break print(f待复核难例数量: {len(hard_cases)})逻辑说明conf0.1放低阈值是为了把模型不确定的框都捞出来0.4是上限高于这个的模型已经很确信不用复核。收集到的文件名交给标注员重点看这批数据的信息量远大于随机抽的图。参数说明阈值区间可以按实际情况调模型越成熟这个区间可以收得越窄。复核完记得把难例按比例混入训练集别全塞进去导致分布偏移。我自己踩过最深的坑是急着上线第一版模型没做难例挖掘就直接部署结果产线上一种罕见的斜向划痕一直漏检返工了一批货才发现。后来养成习惯任何模型上线前必须跑一轮难例挖掘哪怕多花两天。希望帮到你。本文还有配套的精品资源点击获取 YOLOv11工业质检实战：小目标缺陷检测与实时分类部署
简介这份PDF文档面向工业质检领域的技术开发人员与算法工程师围绕YOLOv11目标检测算法系统讲解高精度缺陷检测与实时分类的完整解决方案适合具备一定深度学习基础、希望将目标检测落地到产线质检场景的读者参考。文档共37页为单一PDF文件压缩包约2.04MB支持目录章节跳转与阅读器左侧大纲快速定位查阅体验较为顺畅。内容从工业质检背景与挑战切入依次展开YOLOv11网络结构、训练过程、数据预处理与增强、特征融合与损失函数改进、模型轻量化与硬件加速等关键技术并给出电子芯片、汽车零部件、纺织品三类缺陷检测案例覆盖系统集成部署与未来趋势展望。目前已有73人学习可帮助读者理清从数据标注、模型调优到部署测试的完整链路掌握实时检测与分类的优化思路。1. 从一张机油盖照片说起YOLOv11 工业质检到底在解决什么产线上一个机油盖从注塑机出来表面有没有缩水、划痕、缺料传统做法是靠人眼在强光工位上看。一个人盯 8 小时前两小时准确率能到 95%后面掉到 80% 都算好的而且漏检的件流到总装就是批量返工。YOLOv11 工业质检这套方案要干的事很具体用目标检测把缺陷框出来同时给出类别再让推理速度跟上产线节拍做到实时分类。它适合两类人——一类是手上已经有缺陷图片、想训一个能落地产线模型的算法工程师另一类是做设备集成的需要把检测结果接到 PLC 或分拣机构上。核心矛盾从来不是模型精度不够而是小目标缺陷 实时性 误检率这三者互相拉扯。这篇笔记就按我实际做过的路径把 YOLOv11 从环境配置、数据标注、训练调参到部署推理整条链路拆开讲重点放在小目标优化和实时分类这两个最容易翻车的地方。2. YOLOv11 环境配置与最小可跑通验证别一上来就装全家桶2.1 为什么选 Ultralytics 这套而不是自己搭工业质检场景对框架的要求其实很朴素训练脚本要短、导出格式要多、推理要能塞进 C 或 TensorRT。YOLOv11 在 Ultralytics 仓库里维护model.train()一行就能起训model.export()能直接出 ONNX、TensorRT Engine、OpenVINO这对后面接产线工控机太关键了。自己用 PyTorch 从零搭检测头光是把 NMS 和 letterbox 对齐就要耗掉一周而且导出时各种算子不支持血泪经验就是——能站在成熟仓库上就别造轮子。选型上还有一个现实理由YOLOv11 的网络结构在 neck 部分做了改进对小目标的特征融合比前几代更友好这对轴承缺陷、机油盖划痕这类只占几十个像素的目标很关键。热词里提到的 hcanet、小波缺陷检测本质都是在特征提取阶段做增强但如果你连 baseline 都没跑通加这些模块只会让排查难度翻倍。2.2 环境配置的具体命令与版本约束我一般用 conda 隔离环境Python 锁 3.10CUDA 用 11.8 或 12.1这两个版本和主流 PyTorch 轮子兼容性最好。# 创建独立环境避免和系统里的 torch 打架 conda create -n yolo11 python3.10 -y conda activate yolo11 # 装 PyTorchCUDA 11.8 对应这个 index pip install torch2.1.0 torchvision0.16.0 --index-url https://download.pytorch.org/whl/cu118 # 装 ultralytics不要指定太老的版本 pip install ultralytics # 验证 GPU 是否真的可用 python -c import torch; print(torch.cuda.is_available(), torch.cuda.get_device_name(0))逻辑说明先隔离环境是因为工业项目经常要在一台机器上跑多个版本模型混装必炸。PyTorch 版本和 CUDA 版本必须对齐cu118这个 index 决定了你装的是 GPU 版还是 CPU 版装错了torch.cuda.is_available()会返回 False训练时默默跑在 CPU 上一个 epoch 跑两小时你还以为是数据量大。参数说明torch2.1.0是我验证过和 ultralytics 兼容稳定的版本再新可能遇到算子导出问题。如果你显卡是 40 系CUDA 12.1 也可以把 index 换成cu121。2.3 用官方权重跑一次推理确认链路通在碰自己的数据之前先用预训练权重跑一张图确认从加载模型到出结果整条链路没问题。from ultralytics import YOLO # 加载预训练权重第一次会自动下载 model YOLO(yolo11n.pt) # 对单张图推理saveTrue 会把结果存到 runs/detect/predict results model.predict( sourcetest.jpg, conf0.25, # 置信度阈值工业场景一般先设低一点看召回 iou0.45, # NMS 的 IoU 阈值 saveTrue, device0 # 用第 0 号 GPU ) # 打印每个框的类别和坐标 for box in results[0].boxes: print(box.cls, box.conf, box.xyxy)逻辑说明这一步的价值是排除环境问题。如果这里报错八成是 CUDA 或 torchvision 没装对而不是你的数据有问题。saveTrue会把带框的图存下来肉眼确认框的位置对不对。参数说明conf0.25是通用起点工业质检里如果漏检代价高可以降到 0.1 先看召回iou0.45控制重叠框合并缺陷密集时调低能保留更多框。device0指定 GPU多卡用device[0,1]。3. 数据标注与格式转换缺陷检测的成败八成在这里3.1 工业缺陷数据的三个特殊性工业质检的数据和 COCO 那种自然图像完全不是一回事。第一缺陷样本极度不平衡一批 1000 个件可能只有 20 个有缺陷正负样本比能到 1:50。第二缺陷尺度极小轴承上的点蚀可能就 10×10 像素机油盖的划痕是细长条。第三背景高度一致同一个工位拍的图背景几乎不变模型很容易学到「背景即正常」的捷径换个光照就崩。所以标注策略上我一般要求标注员对每个缺陷框紧贴边缘不要留太多余量因为小目标本身像素就少框大了特征就被背景稀释。类别定义要克制能合并的缺陷类型就合并类别太多会让每个类的样本更少。3.2 从标注文件到 YOLO 格式的转换脚本YOLO 要的是每张图一个 txt每行类别 中心x 中心y 宽 高全部归一化到 0-1。如果你用 LabelImg 标的是 VOC 的 xml需要转。import os import xml.etree.ElementTree as ET def voc_to_yolo(xml_dir, out_dir, classes): # classes 是类别名列表顺序决定类别 id os.makedirs(out_dir, exist_okTrue) for xml_file in os.listdir(xml_dir): if not xml_file.endswith(.xml): continue tree ET.parse(os.path.join(xml_dir, xml_file)) root tree.getroot() size root.find(size) w int(size.find(width).text) h int(size.find(height).text) lines [] for obj in root.iter(object): cls_name obj.find(name).text if cls_name not in classes: continue # 跳过未定义类别避免训练时报错 cls_id classes.index(cls_name) bbox obj.find(bndbox) x1 float(bbox.find(xmin).text) y1 float(bbox.find(ymin).text) x2 float(bbox.find(xmax).text) y2 float(bbox.find(ymax).text) # 归一化并转成中心点宽高 cx (x1 x2) / 2.0 / w cy (y1 y2) / 2.0 / h bw (x2 - x1) / w bh (y2 - y1) / h lines.append(f{cls_id} {cx:.6f} {cy:.6f} {bw:.6f} {bh:.6f}) out_path os.path.join(out_dir, xml_file.replace(.xml, .txt)) with open(out_path, w) as f: f.write(\n.join(lines)) voc_to_yolo(annotations, labels, [scratch, dent, missing])逻辑说明归一化是关键YOLO 不认绝对像素坐标。classes.index把类别名映射成 id顺序必须和后面 data.yaml 里一致否则训练出来的模型类别全错。跳过未定义类别是防止标注员手滑写了个新名字导致训练中断。参数说明cx, cy是框中心点bw, bh是宽高都除以原图宽高做归一化。保留 6 位小数是为了小目标精度小框归一化后数值很小位数不够会丢精度。3.3 data.yaml 的写法与目录结构# data.yaml path: /data/quality_check # 数据集根目录 train: images/train # 训练集图片相对路径 val: images/val # 验证集图片相对路径 nc: 3 # 类别数 names: # 类别名顺序必须和转换脚本一致 0: scratch 1: dent 2: missing目录结构要严格对齐images/train放图labels/train放同名 txtYOLO 会自动去找。很多人翻车是因为图片叫a.jpg标签叫a.txt但放在了别的目录训练时提示找不到标签模型把所有图当负样本训最后啥也检测不出来。提示划分训练验证集时如果同一批次的图背景几乎一样要按批次划分而不是随机划分否则验证集里全是训练集见过的背景指标虚高。4. 训练调参与小目标优化让模型真正看见小缺陷4.1 起训命令与关键参数yolo detect train \ modelyolo11s.pt \ datadata.yaml \ epochs200 \ imgsz640 \ batch16 \ lr00.01 \ lrf0.01 \ patience50 \ augmentTrue \ cacheTrue \ device0逻辑说明modelyolo11s.pt选 s 而不是 n是因为工业缺陷特征弱n 版容量太小容易欠拟合但如果你要部署到边缘设备n 版更合适这是个精度和速度的权衡。patience50是早停50 个 epoch 验证指标不涨就停省时间。参数说明imgsz640是默认输入尺寸小目标多的话可以提到 1280但显存和速度代价大batch16按显存调爆显存就减半lr00.01初始学习率lrf0.01是最终学习率系数cos 衰减到初始的 1%。cacheTrue把图缓存到内存数据量不大时能显著加速。4.2 小目标优化的四个实操手段热词里 yolov11 小目标优化是高频问题我实际用下来有效的就这几招。第一提高输入分辨率。640 提到 1024 或 1280小目标在特征图上的像素变多检测率能明显涨。代价是推理变慢要算清楚产线节拍能不能接受。第二加 P2 检测层。YOLOv11 默认在 P3/P4/P5 三个尺度检测P2 是更高分辨率的浅层特征专门抓小目标。改法是修改模型 yaml在 head 部分加一个 P2 分支。这个改动会让参数量和计算量上升但小目标召回提升明显。第三数据增强针对性调。默认的 mosaic 对小目标有帮助但copy_paste和mixup在缺陷检测里要慎用因为可能把缺陷贴到不合理的位置模型学到错误上下文。我一般把mosaic保留scale范围调窄避免小目标被缩得更小。第四anchor 或损失权重调整。YOLOv11 是 anchor-free 的但可以在损失里给小目标更高权重。这个改动比较深建议先把前三招用满再考虑。4.3 训练过程怎么判断有没有跑偏看runs/detect/train下的results.png重点看三条线box_loss应该稳定下降如果震荡剧烈说明学习率太大mAP50和mAP50-95的差距如果 50 高但 50-95 低说明框的位置不够准小目标定位差cls_loss如果一直不降可能是类别不平衡太严重。还有一个黑匣子式的现象训练集指标很好验证集一塌糊涂这基本是过拟合或者数据泄漏。工业场景里最常见的是同一张图的不同增强版本同时进了训练和验证集指标虚高上线就崩。解决办法是按件号或批次划分数据集别按图片随机分。5. 推理部署与实时分类从模型到产线节拍5.1 导出成部署格式from ultralytics import YOLO model YOLO(runs/detect/train/weights/best.pt) # 导出 ONNX通用性最好 model.export(formatonnx, imgsz640, simplifyTrue, opset12) # 如果有 TensorRT 环境导出 engine 速度最快 model.export(formatengine, imgsz640, halfTrue, device0)逻辑说明ONNX 适合跨平台工控机上用 ONNX Runtime 就能跑TensorRT Engine 是 NVIDIA 显卡上的极致速度但和显卡型号、CUDA 版本绑定换机器要重新导。simplifyTrue会做图优化去掉冗余算子。halfTrue用 FP16速度翻倍精度掉一点点工业场景一般能接受。参数说明opset12兼容性好太新有些推理引擎不支持。imgsz必须和训练时一致否则精度会掉。5.2 实时分类的两种接法标题里的「实时分类」有两层意思。一层是检测框出来之后根据框的类别做分拣决策这个直接在推理结果里读box.cls就行。另一层是整图分类判断这个件是 OK 还是 NG这个可以用检测结果聚合——只要检出任何缺陷框就判 NG。import cv2 from ultralytics import YOLO model YOLO(best.engine) # 用 TensorRT 引擎 cap cv2.VideoCapture(0) # 接工业相机 while True: ret, frame cap.read() if not ret: break results model.predict(frame, conf0.3, verboseFalse) boxes results[0].boxes # 聚合决策有框就是 NG status NG if len(boxes) 0 else OK # 这里可以把 status 通过串口或 PLC 发出去 print(status, len(boxes)) # 保存推理结果图方便追溯 annotated results[0].plot() cv2.imwrite(flog/{int(cap.get(1))}.jpg, annotated)逻辑说明verboseFalse关掉每帧打印否则日志刷屏拖慢速度。results[0].plot()把框画回原图存下来做质量追溯热词里 yolov11 保存推理结果说的就是这个。聚合决策是最简单的实时分类逻辑实际产线可能还要加连续帧确认避免单帧误检触发停机。参数说明conf0.3比训练时略高是为了压误检产线上误检停机比漏检还烦。相机帧率要和推理速度匹配如果推理一帧要 50ms相机就别开 60fps会积压。5.3 速度与精度的平衡表配置输入尺寸精度 mAP50单帧耗时适用场景yolo11n ONNX640基准快边缘设备、节拍快yolo11s TensorRT FP16640略高很快主流工控机yolo11s TensorRT FP161024高中等小目标多、节拍宽松yolo11m TensorRT1280最高慢离线复检这张表是我实际测出来的相对关系具体数字因硬件而异。选型逻辑是先用 s 640 跑通看漏检率不够就提分辨率或换大模型直到满足产线要求为止。6. 避坑与排查那些让模型上线就崩的细节6.1 训练 loss 正常但验证 mAP 为 0现象训练日志里 box_loss 在降但验证集 mAP 一直是 0。原因九成是标签路径或格式不对YOLO 没读到标签把所有框当背景训。解决检查labels/val目录是否存在且和图片同名打开一个 txt 确认是类别 cx cy w h五列且归一化。用yolo detect train时加--verbose看它报的标签数量。6.2 小目标全部漏检现象大缺陷能检出小缺陷一个都框不出来。原因通常是输入分辨率不够小目标在 P3 特征图上只剩一两个像素。解决先把imgsz提到 1024 试如果还不行就加 P2 检测层。另外检查标注框是不是画得太大把背景也框进去了导致特征被稀释。6.3 换一批产品就误检爆炸现象训练集上好好的换了个批次的件背景稍微变了一点误检率飙升。原因是模型学到了背景捷径没学到缺陷本质。解决训练时加更强的颜色抖动和亮度扰动数据里混入不同批次的正常件做负样本让模型知道背景变化不是缺陷。6.4 导出 ONNX 后精度掉一大截现象PyTorch 里推理正常导出 ONNX 后框的位置偏了或类别错了。原因多半是预处理没对齐YOLO 的 letterbox 填充方式在导出后可能和训练时不一致。解决导出时加simplifyTrue推理时用 ultralytics 自己的predict接口而不是自己写预处理或者严格复现 letterbox 的缩放和填充逻辑。6.5 推理速度达不到产线节拍现象单帧耗时 80ms产线要求 30ms。原因可能是没用 TensorRT、没开 FP16、或者后处理 NMS 在 CPU 上跑。解决导出 engine 并开halfTrue把 NMS 放到 GPU 上或者降低输入分辨率。如果还不行考虑模型剪枝或换 n 版。7. 一个提精度的小技巧用检测结果反哺难例挖掘模型训完第一版之后别急着上线。拿它去跑一批没标注的产线图把置信度在 0.1 到 0.4 之间的框导出来这些就是模型「犹豫」的难例。人工复核这批图把漏标的补上、误检的标成负样本再混进训练集重新训一轮。这个循环做两到三轮mAP 通常能再涨几个点比盲目加数据有效得多。from ultralytics import YOLO model YOLO(best.pt) results model.predict(unlabeled_images, conf0.1, saveFalse) # 收集低置信度样本的文件名 hard_cases [] for r in results: for box in r.boxes: if 0.1 box.conf 0.4: hard_cases.append(r.path) break print(f待复核难例数量: {len(hard_cases)})逻辑说明conf0.1放低阈值是为了把模型不确定的框都捞出来0.4是上限高于这个的模型已经很确信不用复核。收集到的文件名交给标注员重点看这批数据的信息量远大于随机抽的图。参数说明阈值区间可以按实际情况调模型越成熟这个区间可以收得越窄。复核完记得把难例按比例混入训练集别全塞进去导致分布偏移。我自己踩过最深的坑是急着上线第一版模型没做难例挖掘就直接部署结果产线上一种罕见的斜向划痕一直漏检返工了一批货才发现。后来养成习惯任何模型上线前必须跑一轮难例挖掘哪怕多花两天。希望帮到你。本文还有配套的精品资源点击获取