简介这份资源是面向深度学习入门者、毕业设计或课程设计学生的YOLOv8钢材表面缺陷检测完整项目包聚焦工业质检场景下的裂纹、凹坑、划痕、锈蚀等缺陷自动识别与分类帮助读者快速搭建可训练、可复现的目标检测流程。压缩包共约2000个文件整体60.81MB其中1801个txt为标注文件187个jpg为缺陷样本图另有5个py脚本负责训练与系统配置、3个pt权重对应不同训练阶段、2个yaml描述数据集与增强策略以及cache和md辅助文件目录结构清晰便于按模块查阅。目前已有80人学习下载。通过该资源读者可获取标注数据、训练脚本、模型权重与运行记录理解参数配置、精度与损失分析思路并据此完成从数据准备到缺陷检测系统落地的完整实践适合作为深度学习图像识别方向的综合项目参考。1. 钢材表面缺陷检测为什么选 YOLOv8一份能跑通的毕设资源做钢材表面缺陷检测这个方向的同学十有八九是被工业质检四个字吸引来的——产线上钢板高速运动人眼根本盯不过来漏检一个划痕可能就是一整批退货。但真到动手才发现公开数据集难找、缺陷样本极度不均衡、标注成本高得离谱光是数据这一关就能卡掉一半人。这份「基于 YOLOv8 的钢材表面缺陷检测设计.zip」解决的正是这个断层它把数据组织、模型训练、推理验证这条链路打包成一套可复现的工程而不是丢给你一个跑不起来的 notebook。适合谁正在做深度学习图像识别方向毕业设计、课程设计或期末大作业的本科生以及想快速摸清 YOLOv8 目标检测完整流程的入门工程师。你不需要从零搭环境但需要一台能跑 CUDA 的机器或者退一步用 CPU 版本先把流程走通。2. 拆开压缩包先看什么目录结构与数据组织逻辑拿到一个深度学习项目压缩包最忌讳的就是上来就python train.py。血泪经验告诉我先花十分钟把目录结构和数据格式摸清楚能省下后面三小时的报错排查。这一章就干这件事。2.1 典型目录长什么样这类钢材缺陷检测项目解压后一般会看到这么几个顶层目录不同作者命名略有差异但职责基本一致steel_defect_yolov8/ ├── datasets/ │ ├── images/ │ │ ├── train/ │ │ └── val/ │ ├── labels/ │ │ ├── train/ │ │ └── val/ │ └── data.yaml ├── models/ │ └── yolov8n.yaml ├── weights/ │ └── yolov8n.pt ├── train.py ├── predict.py ├── val.py └── requirements.txtdatasets是核心images和labels必须严格一一对应——images/train/001.jpg对应labels/train/001.txt文件名相同、扩展名不同。这是 YOLO 系列的硬性约定错一个字符就找不到标签。data.yaml是数据集描述文件告诉训练脚本去哪找图、有几类缺陷、类别叫什么。weights放预训练权重models放网络结构定义。requirements.txt是环境依赖清单先看它再装包别自己瞎猜版本。2.2 data.yaml 到底写了什么这个文件虽小但配错了训练直接崩。典型内容如下# 数据集根路径建议写绝对路径避免相对路径踩坑 path: /home/user/steel_defect_yolov8/datasets train: images/train val: images/val # 缺陷类别数钢材常见的有划痕、凹坑、氧化皮等 nc: 4 # 类别名称顺序必须和标注时的类别 id 对应 names: 0: scratch 1: dent 2: oxide_scale 3: crackpath是数据集根目录train和val是相对path的子路径。nc是类别数量names是类别名到 id 的映射。这里有个高频翻车点标注工具比如 labelme 转 YOLO 格式时生成的类别 id 如果从 1 开始而names从 0 开始训练时类别全错位模型学出来的东西完全是乱的。我一般会先跑一遍脚本统计 labels 里出现过的所有 id确认和names对得上再开训。2.3 标签文件格式与校验YOLO 的标签是纯文本每行一个目标格式为class_id x_center y_center width height后四个值都是归一化到 0~1 的相对坐标不是像素值。很多人从 VOC 的 XML 转过来时忘了除以图像宽高结果坐标全是几百的大数训练 loss 直接爆炸。校验方法很简单写个脚本扫一遍所有标签文件import os label_dir datasets/labels/train for fname in os.listdir(label_dir): with open(os.path.join(label_dir, fname)) as f: for line in f: parts line.strip().split() if len(parts) ! 5: print(f{fname}: 字段数不对 - {line}) continue coords [float(x) for x in parts[1:]] # 归一化坐标必须落在 0~1 之间 if any(c 0 or c 1 for c in coords): print(f{fname}: 坐标越界 - {line})这段脚本干两件事检查每行是不是 5 个字段检查坐标有没有超出 0~1。跑完没有输出说明标签格式基本干净可以进入下一步。有输出就逐条修别带着脏数据训练否则模型精度上不去你还以为是网络结构的问题。3. 环境搭建与训练从 CPU 跑通到 GPU 加速环境配置是劝退重灾区尤其是显卡驱动、CUDA、PyTorch 版本三者之间的兼容玄学。这一章给两条路一条 CPU 保底路线保证你能看到结果一条 GPU 加速路线真正训练时用。3.1 CPU 版本先把流程走通如果你手头暂时没有 N 卡或者只是想先验证代码能不能跑CPU 版本完全够用——训练慢但推理和调试没问题。Ubuntu 20.04 下操作如下# 建虚拟环境别污染系统 Python python3 -m venv yolov8_env source yolov8_env/bin/activate # 装 CPU 版 PyTorch注意 index-url 指向 cpu 版本 pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu # 装 ultralyticsYOLOv8 的官方库 pip install ultralytics # 验证安装 yolo checksyolo checks会打印环境信息重点看 PyTorch 版本和是否检测到 CUDA。CPU 版本这里会显示 CUDA 不可用正常。ultralytics这个库把训练、验证、推理都封装成了命令行工具也支持 Python API后面两种方式都会用到。装完后先拿一张图跑推理确认整条链路通# 用预训练权重跑一张测试图确认模型能加载 yolo predict modelweights/yolov8n.pt sourcedatasets/images/val/001.jpg如果这条命令能输出检测结果图说明环境没问题可以进入训练环节。CPU 训练就把device设成cpu但钢材缺陷数据集通常几千张图CPU 训一个 epoch 可能要几十分钟只适合验证代码逻辑不适合真训。3.2 GPU 版本与训练参数怎么设有 N 卡的话先确认驱动和 CUDA 版本nvidia-smi输出右上角显示 CUDA Version比如 12.1。然后装对应版本的 PyTorch# CUDA 12.1 对应的 PyTorch 安装命令 pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121装完再跑yolo checks这次应该能看到 CUDA 可用。接下来是训练。YOLOv8 的训练入口很简洁yolo detect train \ datadatasets/data.yaml \ modelmodels/yolov8n.yaml \ pretrainedweights/yolov8n.pt \ epochs100 \ imgsz640 \ batch16 \ device0 \ projectruns/train \ namesteel_exp1逐个参数说清楚data指向数据集配置文件model是网络结构yolov8n是最小的 nano 版本速度快精度低钢材缺陷检测如果缺陷特征细微建议换yolov8s或yolov8mpretrained加载预训练权重迁移学习能大幅缩短收敛时间epochs是训练轮数100 起步看 loss 曲线决定要不要加imgsz是输入图像尺寸640 是默认值缺陷特别小可以提到 1280但显存占用翻倍batch是批大小显存不够就往下调8 甚至 4 都行device0指定第一块 GPUCPU 就写cpu。project和name控制输出目录训练日志、权重、曲线图都会存到runs/train/steel_exp1下面。训练启动后终端会实时打印每个 epoch 的 loss 和 mAP。重点盯两个指标box_loss和mAP50。box_loss持续下降说明模型在学如果震荡不降多半是学习率太大或者数据有问题mAP50是 IoU 阈值 0.5 时的平均精度钢材缺陷检测一般能到 0.7 以上算不错。训练结束后runs/train/steel_exp1/weights/下会有best.pt和last.ptbest.pt是验证集表现最好的权重推理用这个。3.3 损失函数曲线怎么读训练完runs/train/steel_exp1/下会有results.png包含训练和验证的 loss、precision、recall、mAP 曲线。这张图是判断模型状态的直接依据。正常情况训练 loss 和验证 loss 同步下降最后趋于平稳如果训练 loss 降但验证 loss 上升是过拟合需要加数据增强或减模型复杂度如果两条都居高不下是欠拟合加 epoch 或换更大模型。钢材缺陷检测常见的问题是某些类别样本太少导致该类别的 AP 明显低于其他类这时候要么补样本要么在训练时给类别加权。4. 推理、验证与模型导出让训练结果真正可用训练出best.pt只是半成品得验证它在没见过的图上表现如何再决定要不要导出成部署格式。这一章讲验证和导出的具体操作。4.1 用 val.py 跑验证集指标训练时其实已经跑过验证但单独再跑一次可以指定不同权重和参数yolo detect val \ modelruns/train/steel_exp1/weights/best.pt \ datadatasets/data.yaml \ imgsz640 \ batch16 \ device0输出会列出每个类别的 precision、recall、mAP50、mAP50-95。mAP50-95比mAP50更严格因为它在多个 IoU 阈值下取平均。钢材缺陷检测里如果mAP50高但mAP50-95低说明模型能检测到缺陷但框的位置不够准可能需要调整锚框或加更多定位损失权重。验证完还会生成混淆矩阵和 PR 曲线存在runs/val/下混淆矩阵能直观看出哪两类缺陷容易混——比如划痕和裂纹在某些光照下确实难分这是数据本身的问题不是模型不行。4.2 单张图和批量推理验证指标是宏观的实际用的时候还是得看图。单张推理yolo predict \ modelruns/train/steel_exp1/weights/best.pt \ sourcedatasets/images/val/001.jpg \ conf0.25 \ saveTrueconf是置信度阈值低于这个值的检测框会被过滤掉。钢材缺陷检测建议从 0.25 起步漏检多就降到 0.1误检多就提到 0.5。saveTrue会把带框的图存到runs/detect/下。批量推理把source换成目录路径即可yolo predict \ modelruns/train/steel_exp1/weights/best.pt \ sourcedatasets/images/val/ \ conf0.25 \ saveTrue批量跑完重点看两类图漏检的有缺陷但没框和误检的没缺陷但框了。漏检多说明模型对该类缺陷不敏感可能是样本太少或特征太弱误检多说明模型把正常纹理当成了缺陷需要加负样本。这两种情况在钢材数据集里都很常见因为钢板表面的氧化皮、反光、水渍本身就容易干扰。4.3 导出 ONNX 和 TensorRT如果只是交毕设best.pt足够了。但如果想往部署方向走导出 ONNX 是第一步yolo export \ modelruns/train/steel_exp1/weights/best.pt \ formatonnx \ imgsz640 \ opset12 \ simplifyTrueformat指定导出格式opset是 ONNX 算子集版本12 兼容性较好simplify会简化计算图减小模型体积。导出后的.onnx文件可以用 Netron 可视化结构也可以用 onnxruntime 推理。如果要上边缘设备比如 RK3588 这类板子通常还需要进一步转成 RKNN 格式但那是另一个话题了这份资源主要覆盖到训练和 ONNX 导出。导出时注意imgsz要和训练时一致否则精度会掉。5. 避坑与常见问题那些让我重训三次的坑这一章全是真金白银换来的教训每条都按「现象 → 原因 → 解决」写对号入座能省大量时间。5.1 训练 loss 为 nan现象训练启动后几个 batchbox_loss变成nan之后一直不降。原因最常见的是标签坐标越界或格式错误导致损失计算出负值或除零其次是学习率设得太大梯度爆炸。解决先跑 2.3 节的标签校验脚本确认所有坐标在 0~1 之间、每行 5 个字段。标签没问题就把学习率从默认的 0.01 降到 0.001 试试YOLOv8 默认用 AdamW 优化器对学习率比 SGD 敏感。5.2 显存不足 OOM现象训练刚开始就报CUDA out of memory。原因batch太大、imgsz太大或者显卡本身显存小比如 6GB 的 GTX 1660 Ti。解决优先降batch从 16 降到 8 再到 4还不够就降imgsz从 640 降到 512 甚至 416。另外可以开梯度累积模拟大 batchbatch4配合accumulate4效果接近batch16但显存占用按 4 算。GTX 1660 Ti 跑yolov8nimgsz640batch8基本是极限再大就爆。5.3 验证集 mAP 始终上不去现象训练 loss 正常下降但mAP50卡在 0.3 左右不动。原因数据层面问题居多——类别不均衡某类缺陷只有几十张图、标注质量差框不准或漏标、训练集和验证集分布差异大。解决先统计每个类别的样本数少于 100 张的类别考虑过采样或加数据增强再抽查标注用yolo predict跑几张训练图看模型学到的框和标注框差多少最后检查训练集和验证集是不是随机划分的如果验证集全是某种光照条件下的图模型泛化肯定差。5.4 推理时检测框重叠严重现象同一处缺陷被框了好几次框叠在一起。原因NMS非极大值抑制的 IoU 阈值设得太高或者模型对同一目标输出了多个高置信度框。解决推理时加iou参数控制 NMS 阈值默认 0.7降到 0.5 能减少重叠框yolo predict modelbest.pt sourcetest.jpg iou0.5 conf0.25如果降了还重叠说明模型训练时正样本分配有问题可能需要检查 anchor 配置或换更小的模型。5.5 换数据集后类别名对不上现象用自己的数据训练推理时类别名显示成class_0、class_1而不是实际缺陷名。原因data.yaml里的names没改或者改了但 id 和标注文件里的对不上。解决打开data.yaml确认nc等于实际类别数names的 id 从 0 开始连续且和标签文件里的class_id一一对应。改完重新训练类别名是在训练时写入权重文件的光改 yaml 不重训没用。6. 把这份资源用出最大价值迁移到自己的数据与进阶调参这份资源最大的价值不是让你跑通一个现成的钢材缺陷检测而是给你一套可迁移的模板。把datasets换成你自己的数据、改data.yaml的类别配置、重新训练就能套用到其他表面缺陷场景——铝板划痕、PCB 焊点缺陷、织物疵点流程完全一样。我一般会先把这份资源在钢材数据上跑通一遍确认环境、训练、推理每个环节都正常再动自己的数据。这样出问题时能快速定位是环境问题还是数据问题。迁移时最容易翻车的是标注格式。如果你用 labelme 标数据它默认输出 JSON需要转成 YOLO 的 txt 格式。转换脚本核心逻辑是读 JSON 里的shapes取每个多边形的外接矩形再归一化import json import os from PIL import Image def labelme_to_yolo(json_path, output_dir, class_map): with open(json_path) as f: data json.load(f) img_w data[imageWidth] img_h data[imageHeight] lines [] for shape in data[shapes]: label shape[label] if label not in class_map: continue cls_id class_map[label] points shape[points] xs [p[0] for p in points] ys [p[1] for p in points] # 外接矩形转 YOLO 中心点宽高格式 x_center (min(xs) max(xs)) / 2 / img_w y_center (min(ys) max(ys)) / 2 / img_h width (max(xs) - min(xs)) / img_w height (max(ys) - min(ys)) / img_h lines.append(f{cls_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}) out_name os.path.splitext(os.path.basename(json_path))[0] .txt with open(os.path.join(output_dir, out_name), w) as f: f.write(\n.join(lines))class_map是类别名到 id 的字典必须和data.yaml里的names一致。转换完同样跑一遍 2.3 节的校验脚本确认坐标都在 0~1 之间。这个脚本我改过很多次最早版本忘了归一化训出来的模型框全在图像左上角排查了半天才发现是坐标没除宽高。调参方面钢材缺陷检测有几个经验值学习率用0.001起步比默认的0.01稳数据增强里mosaic对小缺陷有帮助但别开太大mosaic0.5左右patience设 20 到 30连续这么多 epoch 验证指标不升就早停省时间。如果缺陷特别小比如几个像素的裂纹把imgsz提到 1024 或 1280同时batch降到 4精度提升明显但训练慢一倍。从那以后我每次拿到新的检测数据集都强制先跑一遍标签校验和类别统计再开训。这个习惯帮我省下了至少三次重训的时间。希望这份资源和你自己的数据能顺利对上帮到你。本文还有配套的精品资源点击获取 基于YOLOv8的钢材表面缺陷检测：从数据组织到模型部署的完整实战指南
简介这份资源是面向深度学习入门者、毕业设计或课程设计学生的YOLOv8钢材表面缺陷检测完整项目包聚焦工业质检场景下的裂纹、凹坑、划痕、锈蚀等缺陷自动识别与分类帮助读者快速搭建可训练、可复现的目标检测流程。压缩包共约2000个文件整体60.81MB其中1801个txt为标注文件187个jpg为缺陷样本图另有5个py脚本负责训练与系统配置、3个pt权重对应不同训练阶段、2个yaml描述数据集与增强策略以及cache和md辅助文件目录结构清晰便于按模块查阅。目前已有80人学习下载。通过该资源读者可获取标注数据、训练脚本、模型权重与运行记录理解参数配置、精度与损失分析思路并据此完成从数据准备到缺陷检测系统落地的完整实践适合作为深度学习图像识别方向的综合项目参考。1. 钢材表面缺陷检测为什么选 YOLOv8一份能跑通的毕设资源做钢材表面缺陷检测这个方向的同学十有八九是被工业质检四个字吸引来的——产线上钢板高速运动人眼根本盯不过来漏检一个划痕可能就是一整批退货。但真到动手才发现公开数据集难找、缺陷样本极度不均衡、标注成本高得离谱光是数据这一关就能卡掉一半人。这份「基于 YOLOv8 的钢材表面缺陷检测设计.zip」解决的正是这个断层它把数据组织、模型训练、推理验证这条链路打包成一套可复现的工程而不是丢给你一个跑不起来的 notebook。适合谁正在做深度学习图像识别方向毕业设计、课程设计或期末大作业的本科生以及想快速摸清 YOLOv8 目标检测完整流程的入门工程师。你不需要从零搭环境但需要一台能跑 CUDA 的机器或者退一步用 CPU 版本先把流程走通。2. 拆开压缩包先看什么目录结构与数据组织逻辑拿到一个深度学习项目压缩包最忌讳的就是上来就python train.py。血泪经验告诉我先花十分钟把目录结构和数据格式摸清楚能省下后面三小时的报错排查。这一章就干这件事。2.1 典型目录长什么样这类钢材缺陷检测项目解压后一般会看到这么几个顶层目录不同作者命名略有差异但职责基本一致steel_defect_yolov8/ ├── datasets/ │ ├── images/ │ │ ├── train/ │ │ └── val/ │ ├── labels/ │ │ ├── train/ │ │ └── val/ │ └── data.yaml ├── models/ │ └── yolov8n.yaml ├── weights/ │ └── yolov8n.pt ├── train.py ├── predict.py ├── val.py └── requirements.txtdatasets是核心images和labels必须严格一一对应——images/train/001.jpg对应labels/train/001.txt文件名相同、扩展名不同。这是 YOLO 系列的硬性约定错一个字符就找不到标签。data.yaml是数据集描述文件告诉训练脚本去哪找图、有几类缺陷、类别叫什么。weights放预训练权重models放网络结构定义。requirements.txt是环境依赖清单先看它再装包别自己瞎猜版本。2.2 data.yaml 到底写了什么这个文件虽小但配错了训练直接崩。典型内容如下# 数据集根路径建议写绝对路径避免相对路径踩坑 path: /home/user/steel_defect_yolov8/datasets train: images/train val: images/val # 缺陷类别数钢材常见的有划痕、凹坑、氧化皮等 nc: 4 # 类别名称顺序必须和标注时的类别 id 对应 names: 0: scratch 1: dent 2: oxide_scale 3: crackpath是数据集根目录train和val是相对path的子路径。nc是类别数量names是类别名到 id 的映射。这里有个高频翻车点标注工具比如 labelme 转 YOLO 格式时生成的类别 id 如果从 1 开始而names从 0 开始训练时类别全错位模型学出来的东西完全是乱的。我一般会先跑一遍脚本统计 labels 里出现过的所有 id确认和names对得上再开训。2.3 标签文件格式与校验YOLO 的标签是纯文本每行一个目标格式为class_id x_center y_center width height后四个值都是归一化到 0~1 的相对坐标不是像素值。很多人从 VOC 的 XML 转过来时忘了除以图像宽高结果坐标全是几百的大数训练 loss 直接爆炸。校验方法很简单写个脚本扫一遍所有标签文件import os label_dir datasets/labels/train for fname in os.listdir(label_dir): with open(os.path.join(label_dir, fname)) as f: for line in f: parts line.strip().split() if len(parts) ! 5: print(f{fname}: 字段数不对 - {line}) continue coords [float(x) for x in parts[1:]] # 归一化坐标必须落在 0~1 之间 if any(c 0 or c 1 for c in coords): print(f{fname}: 坐标越界 - {line})这段脚本干两件事检查每行是不是 5 个字段检查坐标有没有超出 0~1。跑完没有输出说明标签格式基本干净可以进入下一步。有输出就逐条修别带着脏数据训练否则模型精度上不去你还以为是网络结构的问题。3. 环境搭建与训练从 CPU 跑通到 GPU 加速环境配置是劝退重灾区尤其是显卡驱动、CUDA、PyTorch 版本三者之间的兼容玄学。这一章给两条路一条 CPU 保底路线保证你能看到结果一条 GPU 加速路线真正训练时用。3.1 CPU 版本先把流程走通如果你手头暂时没有 N 卡或者只是想先验证代码能不能跑CPU 版本完全够用——训练慢但推理和调试没问题。Ubuntu 20.04 下操作如下# 建虚拟环境别污染系统 Python python3 -m venv yolov8_env source yolov8_env/bin/activate # 装 CPU 版 PyTorch注意 index-url 指向 cpu 版本 pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu # 装 ultralyticsYOLOv8 的官方库 pip install ultralytics # 验证安装 yolo checksyolo checks会打印环境信息重点看 PyTorch 版本和是否检测到 CUDA。CPU 版本这里会显示 CUDA 不可用正常。ultralytics这个库把训练、验证、推理都封装成了命令行工具也支持 Python API后面两种方式都会用到。装完后先拿一张图跑推理确认整条链路通# 用预训练权重跑一张测试图确认模型能加载 yolo predict modelweights/yolov8n.pt sourcedatasets/images/val/001.jpg如果这条命令能输出检测结果图说明环境没问题可以进入训练环节。CPU 训练就把device设成cpu但钢材缺陷数据集通常几千张图CPU 训一个 epoch 可能要几十分钟只适合验证代码逻辑不适合真训。3.2 GPU 版本与训练参数怎么设有 N 卡的话先确认驱动和 CUDA 版本nvidia-smi输出右上角显示 CUDA Version比如 12.1。然后装对应版本的 PyTorch# CUDA 12.1 对应的 PyTorch 安装命令 pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121装完再跑yolo checks这次应该能看到 CUDA 可用。接下来是训练。YOLOv8 的训练入口很简洁yolo detect train \ datadatasets/data.yaml \ modelmodels/yolov8n.yaml \ pretrainedweights/yolov8n.pt \ epochs100 \ imgsz640 \ batch16 \ device0 \ projectruns/train \ namesteel_exp1逐个参数说清楚data指向数据集配置文件model是网络结构yolov8n是最小的 nano 版本速度快精度低钢材缺陷检测如果缺陷特征细微建议换yolov8s或yolov8mpretrained加载预训练权重迁移学习能大幅缩短收敛时间epochs是训练轮数100 起步看 loss 曲线决定要不要加imgsz是输入图像尺寸640 是默认值缺陷特别小可以提到 1280但显存占用翻倍batch是批大小显存不够就往下调8 甚至 4 都行device0指定第一块 GPUCPU 就写cpu。project和name控制输出目录训练日志、权重、曲线图都会存到runs/train/steel_exp1下面。训练启动后终端会实时打印每个 epoch 的 loss 和 mAP。重点盯两个指标box_loss和mAP50。box_loss持续下降说明模型在学如果震荡不降多半是学习率太大或者数据有问题mAP50是 IoU 阈值 0.5 时的平均精度钢材缺陷检测一般能到 0.7 以上算不错。训练结束后runs/train/steel_exp1/weights/下会有best.pt和last.ptbest.pt是验证集表现最好的权重推理用这个。3.3 损失函数曲线怎么读训练完runs/train/steel_exp1/下会有results.png包含训练和验证的 loss、precision、recall、mAP 曲线。这张图是判断模型状态的直接依据。正常情况训练 loss 和验证 loss 同步下降最后趋于平稳如果训练 loss 降但验证 loss 上升是过拟合需要加数据增强或减模型复杂度如果两条都居高不下是欠拟合加 epoch 或换更大模型。钢材缺陷检测常见的问题是某些类别样本太少导致该类别的 AP 明显低于其他类这时候要么补样本要么在训练时给类别加权。4. 推理、验证与模型导出让训练结果真正可用训练出best.pt只是半成品得验证它在没见过的图上表现如何再决定要不要导出成部署格式。这一章讲验证和导出的具体操作。4.1 用 val.py 跑验证集指标训练时其实已经跑过验证但单独再跑一次可以指定不同权重和参数yolo detect val \ modelruns/train/steel_exp1/weights/best.pt \ datadatasets/data.yaml \ imgsz640 \ batch16 \ device0输出会列出每个类别的 precision、recall、mAP50、mAP50-95。mAP50-95比mAP50更严格因为它在多个 IoU 阈值下取平均。钢材缺陷检测里如果mAP50高但mAP50-95低说明模型能检测到缺陷但框的位置不够准可能需要调整锚框或加更多定位损失权重。验证完还会生成混淆矩阵和 PR 曲线存在runs/val/下混淆矩阵能直观看出哪两类缺陷容易混——比如划痕和裂纹在某些光照下确实难分这是数据本身的问题不是模型不行。4.2 单张图和批量推理验证指标是宏观的实际用的时候还是得看图。单张推理yolo predict \ modelruns/train/steel_exp1/weights/best.pt \ sourcedatasets/images/val/001.jpg \ conf0.25 \ saveTrueconf是置信度阈值低于这个值的检测框会被过滤掉。钢材缺陷检测建议从 0.25 起步漏检多就降到 0.1误检多就提到 0.5。saveTrue会把带框的图存到runs/detect/下。批量推理把source换成目录路径即可yolo predict \ modelruns/train/steel_exp1/weights/best.pt \ sourcedatasets/images/val/ \ conf0.25 \ saveTrue批量跑完重点看两类图漏检的有缺陷但没框和误检的没缺陷但框了。漏检多说明模型对该类缺陷不敏感可能是样本太少或特征太弱误检多说明模型把正常纹理当成了缺陷需要加负样本。这两种情况在钢材数据集里都很常见因为钢板表面的氧化皮、反光、水渍本身就容易干扰。4.3 导出 ONNX 和 TensorRT如果只是交毕设best.pt足够了。但如果想往部署方向走导出 ONNX 是第一步yolo export \ modelruns/train/steel_exp1/weights/best.pt \ formatonnx \ imgsz640 \ opset12 \ simplifyTrueformat指定导出格式opset是 ONNX 算子集版本12 兼容性较好simplify会简化计算图减小模型体积。导出后的.onnx文件可以用 Netron 可视化结构也可以用 onnxruntime 推理。如果要上边缘设备比如 RK3588 这类板子通常还需要进一步转成 RKNN 格式但那是另一个话题了这份资源主要覆盖到训练和 ONNX 导出。导出时注意imgsz要和训练时一致否则精度会掉。5. 避坑与常见问题那些让我重训三次的坑这一章全是真金白银换来的教训每条都按「现象 → 原因 → 解决」写对号入座能省大量时间。5.1 训练 loss 为 nan现象训练启动后几个 batchbox_loss变成nan之后一直不降。原因最常见的是标签坐标越界或格式错误导致损失计算出负值或除零其次是学习率设得太大梯度爆炸。解决先跑 2.3 节的标签校验脚本确认所有坐标在 0~1 之间、每行 5 个字段。标签没问题就把学习率从默认的 0.01 降到 0.001 试试YOLOv8 默认用 AdamW 优化器对学习率比 SGD 敏感。5.2 显存不足 OOM现象训练刚开始就报CUDA out of memory。原因batch太大、imgsz太大或者显卡本身显存小比如 6GB 的 GTX 1660 Ti。解决优先降batch从 16 降到 8 再到 4还不够就降imgsz从 640 降到 512 甚至 416。另外可以开梯度累积模拟大 batchbatch4配合accumulate4效果接近batch16但显存占用按 4 算。GTX 1660 Ti 跑yolov8nimgsz640batch8基本是极限再大就爆。5.3 验证集 mAP 始终上不去现象训练 loss 正常下降但mAP50卡在 0.3 左右不动。原因数据层面问题居多——类别不均衡某类缺陷只有几十张图、标注质量差框不准或漏标、训练集和验证集分布差异大。解决先统计每个类别的样本数少于 100 张的类别考虑过采样或加数据增强再抽查标注用yolo predict跑几张训练图看模型学到的框和标注框差多少最后检查训练集和验证集是不是随机划分的如果验证集全是某种光照条件下的图模型泛化肯定差。5.4 推理时检测框重叠严重现象同一处缺陷被框了好几次框叠在一起。原因NMS非极大值抑制的 IoU 阈值设得太高或者模型对同一目标输出了多个高置信度框。解决推理时加iou参数控制 NMS 阈值默认 0.7降到 0.5 能减少重叠框yolo predict modelbest.pt sourcetest.jpg iou0.5 conf0.25如果降了还重叠说明模型训练时正样本分配有问题可能需要检查 anchor 配置或换更小的模型。5.5 换数据集后类别名对不上现象用自己的数据训练推理时类别名显示成class_0、class_1而不是实际缺陷名。原因data.yaml里的names没改或者改了但 id 和标注文件里的对不上。解决打开data.yaml确认nc等于实际类别数names的 id 从 0 开始连续且和标签文件里的class_id一一对应。改完重新训练类别名是在训练时写入权重文件的光改 yaml 不重训没用。6. 把这份资源用出最大价值迁移到自己的数据与进阶调参这份资源最大的价值不是让你跑通一个现成的钢材缺陷检测而是给你一套可迁移的模板。把datasets换成你自己的数据、改data.yaml的类别配置、重新训练就能套用到其他表面缺陷场景——铝板划痕、PCB 焊点缺陷、织物疵点流程完全一样。我一般会先把这份资源在钢材数据上跑通一遍确认环境、训练、推理每个环节都正常再动自己的数据。这样出问题时能快速定位是环境问题还是数据问题。迁移时最容易翻车的是标注格式。如果你用 labelme 标数据它默认输出 JSON需要转成 YOLO 的 txt 格式。转换脚本核心逻辑是读 JSON 里的shapes取每个多边形的外接矩形再归一化import json import os from PIL import Image def labelme_to_yolo(json_path, output_dir, class_map): with open(json_path) as f: data json.load(f) img_w data[imageWidth] img_h data[imageHeight] lines [] for shape in data[shapes]: label shape[label] if label not in class_map: continue cls_id class_map[label] points shape[points] xs [p[0] for p in points] ys [p[1] for p in points] # 外接矩形转 YOLO 中心点宽高格式 x_center (min(xs) max(xs)) / 2 / img_w y_center (min(ys) max(ys)) / 2 / img_h width (max(xs) - min(xs)) / img_w height (max(ys) - min(ys)) / img_h lines.append(f{cls_id} {x_center:.6f} {y_center:.6f} {width:.6f} {height:.6f}) out_name os.path.splitext(os.path.basename(json_path))[0] .txt with open(os.path.join(output_dir, out_name), w) as f: f.write(\n.join(lines))class_map是类别名到 id 的字典必须和data.yaml里的names一致。转换完同样跑一遍 2.3 节的校验脚本确认坐标都在 0~1 之间。这个脚本我改过很多次最早版本忘了归一化训出来的模型框全在图像左上角排查了半天才发现是坐标没除宽高。调参方面钢材缺陷检测有几个经验值学习率用0.001起步比默认的0.01稳数据增强里mosaic对小缺陷有帮助但别开太大mosaic0.5左右patience设 20 到 30连续这么多 epoch 验证指标不升就早停省时间。如果缺陷特别小比如几个像素的裂纹把imgsz提到 1024 或 1280同时batch降到 4精度提升明显但训练慢一倍。从那以后我每次拿到新的检测数据集都强制先跑一遍标签校验和类别统计再开训。这个习惯帮我省下了至少三次重训的时间。希望这份资源和你自己的数据能顺利对上帮到你。本文还有配套的精品资源点击获取