简介本资源面向计算机视觉学习者与智能交通方向开发者提供一套基于Yolov8的道路病害目标检测完整项目覆盖横向裂缝、纵向裂缝、块状裂缝、龟裂、坑槽及多种修补类病害的识别任务适合课程大作业、毕业设计或工程原型验证。压缩包共549个文件约1.76MB以345个md文档、124个py脚本、26个yaml与22个yml配置为主辅以xml、txt、sh及少量前端与部署文件涵盖数据预处理、模型训练、权重保存与加载等完整流程。已有618人学习下载。项目内含源码、模型、数据集说明与项目说明读者可据此复现从数据集准备、参数配置、训练调优到mAP、召回率、精确率评估的全过程并理解如何将模型部署到无人机巡检、车载摄像头等实时场景为智能交通与智慧城市应用提供可迁移的技术参考。1. 道路病害检测从 YOLOv8 开始一份大作业源码到底能跑出什么道路病害目标检测这件事真正做过的人都知道难点从来不在模型本身而在数据。坑槽、裂缝、龟裂、沉陷这几类病害在真实道路影像里尺度差异极大——一条细裂缝可能只占十几个像素一块坑槽却能横跨半个画面。YOLOv8 作为单阶段检测器在这个任务上算是目前性价比最高的选择训练快、部署轻、对小目标也有一定容忍度。这份大作业打包了源码、模型、数据集说明和项目说明本质上是一套「从零跑通道路病害检测」的最小闭环。适合谁课程设计要交东西的学生、想入门工业缺陷检测的工程师、以及手里已经有一批道路巡检图但不知道怎么下手的人。下面我按自己复现这类项目的顺序把每一步拆开讲。2. 道路病害数据集怎么准备从原始标注到 YOLO 格式2.1 先搞清楚你的数据长什么样道路病害数据集通常来自车载巡检设备或手机拍摄分辨率从 1920×1080 到 4000×3000 都有。拿到手第一件事不是急着训练而是统计三件事类别分布、标注格式、图像尺寸分布。类别分布决定你要不要做重采样标注格式决定你要不要写转换脚本图像尺寸分布决定你 imgsz 设多大。常见做法是先用一段脚本把数据集扫一遍输出每个类别的框数量和图像尺寸直方图。这一步花十分钟能省掉后面几小时的无效训练。import os import cv2 from collections import Counter # 假设标注是 YOLO txt 格式每行: class_id cx cy w h label_dir datasets/road_damage/labels/train img_dir datasets/road_damage/images/train cls_counter Counter() size_counter Counter() for txt in os.listdir(label_dir): if not txt.endswith(.txt): continue with open(os.path.join(label_dir, txt)) as f: for line in f: cls_id int(line.split()[0]) cls_counter[cls_id] 1 # 对应图像尺寸 img_path os.path.join(img_dir, txt.replace(.txt, .jpg)) if os.path.exists(img_path): h, w cv2.imread(img_path).shape[:2] size_counter[(w, h)] 1 print(类别分布:, cls_counter) print(尺寸分布 Top5:, size_counter.most_common(5))这段脚本的逻辑很直白遍历标签文件统计类别再读对应图像拿尺寸。参数上唯一要注意的是img_dir和label_dir的对应关系——YOLO 要求图像和标签同名不同后缀如果你的数据集不是这个结构先重命名对齐。类别分布如果严重倾斜比如裂缝占 80%沉陷只占 2%后面训练时要么加类别权重要么对少样本类做增强。2.2 标注格式转换的四个边界坑很多道路病害数据集原始标注是 VOC XML 或 LabelMe JSON转成 YOLO txt 时最容易翻车的地方有四个第一坐标归一化。YOLO 的 cx cy w h 都是相对图像宽高的比例不是绝对像素。转换时除以宽高就完事但要注意浮点精度保留六位小数足够。第二越界框。标注时手抖画到图像外面是常事转换后 cx 可能小于 0 或大于 1。这种框直接裁到边界别留着否则训练时数据增强会报错。第三空标签文件。有些图像没有病害对应 txt 是空的。YOLOv8 默认会跳过空标签但如果你开了rect训练模式空文件可能导致 batch 组装异常。建议要么删掉空图要么在配置里显式处理。第四类别 ID 从 0 开始。VOC 的类别名转 YOLO 时一定要建一个固定的 name-to-id 映射表别用set去重后随机排序否则每次转换 ID 都在变模型根本学不到东西。import xml.etree.ElementTree as ET import os # 固定类别映射顺序一旦定下就不要改 CLASS_MAP {crack: 0, pothole: 1, alligator: 2, subsidence: 3} def voc_to_yolo(xml_path, img_w, img_h, out_path): tree ET.parse(xml_path) root tree.getroot() lines [] for obj in root.findall(object): name obj.find(name).text.strip() if name not in CLASS_MAP: continue cls_id CLASS_MAP[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) # 裁剪越界坐标 x1, x2 max(0, x1), min(img_w, x2) y1, y2 max(0, y1), min(img_h, y2) if x2 x1 or y2 y1: continue cx (x1 x2) / 2 / img_w cy (y1 y2) / 2 / img_h w (x2 - x1) / img_w h (y2 - y1) / img_h lines.append(f{cls_id} {cx:.6f} {cy:.6f} {w:.6f} {h:.6f}) with open(out_path, w) as f: f.write(\n.join(lines))这段代码的关键在CLASS_MAP和越界裁剪。CLASS_MAP必须手动维护别偷懒用自动排序。裁剪那两行是后悔药不加的话后面增强阶段会随机崩。转换完记得抽查几张可视化确认框位置没偏。2.3 数据集划分与 data.yaml 配置YOLOv8 要求一个data.yaml描述数据集路径和类别。典型结构是 train/val/test 三个子集比例 7:2:1 或 8:1:1。道路病害数据如果来自连续视频帧千万别随机划分——相邻帧几乎一样随机划分会导致验证集泄漏指标虚高。正确做法是按视频段或采集日期划分。# data.yaml path: /home/user/road_damage train: images/train val: images/val test: images/test names: 0: crack 1: pothole 2: alligator 3: subsidencepath写绝对路径最稳相对路径在不同工作目录下容易找不到。names的顺序必须和转换时的CLASS_MAP完全一致错一个就是灾难。3. YOLOv8 训练道路病害模型参数怎么设、曲线怎么看3.1 环境配置与最小训练命令环境这块YOLOv8 依赖 ultralytics 包Python 3.8 以上都行。如果你用 GPU先确认 CUDA 和 PyTorch 版本匹配。GTX 1660 Ti 这种 6G 显存的卡跑 yolov8n 或 yolov8s 没问题yolov8m 就要降 batch 了。pip install ultralytics yolo detect train \ datadata.yaml \ modelyolov8n.pt \ epochs100 \ imgsz640 \ batch16 \ device0 \ projectruns/road \ nameexp1这是最小可跑命令。modelyolov8n.pt会用预训练权重道路病害数据量不大时迁移学习比从头训快得多。imgsz640是默认值如果你的裂缝特别细可以提到 1024但显存占用会翻倍。batch16在 6G 卡上跑 640 尺寸刚好爆显存就降到 8。3.2 道路病害场景下的关键参数默认参数能跑但想跑好这几个参数必须调imgsz裂缝类病害建议 1024坑槽类 640 够用。判断依据是标注框的最小边长——如果大量框小于 20 像素就得提分辨率。batch显存允许范围内越大越好但别超过 32否则学习率要跟着调。小数据集用 8 或 16 更稳。lr0初始学习率默认 0.01。道路病害数据如果只有几千张降到 0.001 更稳避免早期震荡。patience早停轮数默认 50。小数据集设 20 就够省时间。close_mosaic最后多少轮关闭 mosaic 增强默认 10。道路病害的裂缝在 mosaic 拼接后容易断裂建议设 20让模型最后阶段看到真实分布。yolo detect train \ datadata.yaml \ modelyolov8s.pt \ epochs200 \ imgsz1024 \ batch8 \ lr00.001 \ patience20 \ close_mosaic20 \ device0这套参数是我在类似任务上试出来的折中值。imgsz 提到 1024 后小裂缝召回率能涨 5 到 8 个点但训练时间翻倍。如果时间紧640 也能出活只是细裂缝漏检多。3.3 损失曲线怎么读翻车信号识别训练启动后runs/road/exp1/下会生成results.csv和损失曲线图。重点看三条线box_loss、cls_loss、dfl_loss。正常情况三条都单调下降最后趋于平缓。翻车信号一box_loss 下降但 cls_loss 不降。说明模型能定位但分不清类别通常是类别不平衡或标注噪声。解决方法是检查标注或者给少样本类加权。翻车信号二val_loss 先降后升。典型过拟合加数据增强或减模型容量。道路病害数据增强推荐用mosaic0.5、flipud0.5、hsv_h0.015别开太大旋转裂缝方向是重要特征。翻车信号三mAP 卡在 0.3 以下不动。先别怀疑模型去可视化验证集预测结果。八成是标注格式错了或者类别 ID 对不上。import pandas as pd import matplotlib.pyplot as plt df pd.read_csv(runs/road/exp1/results.csv) df.columns df.columns.str.strip() fig, axes plt.subplots(1, 2, figsize(12, 4)) axes[0].plot(df[epoch], df[train/box_loss], labelbox) axes[0].plot(df[epoch], df[train/cls_loss], labelcls) axes[0].plot(df[epoch], df[train/dfl_loss], labeldfl) axes[0].legend() axes[0].set_title(Train Loss) axes[1].plot(df[epoch], df[metrics/mAP50(B)], labelmAP50) axes[1].plot(df[epoch], df[metrics/mAP50-95(B)], labelmAP50-95) axes[1].legend() axes[1].set_title(Val mAP) plt.savefig(curve.png)这段脚本把训练日志画成两张图。注意列名可能有空格strip()先处理掉。mAP50 到 0.6 以上算可用0.75 以上算不错。道路病害因为标注主观性强别追求 0.9不现实。4. 模型验证与推理mAP 之外还要看什么4.1 验证集评估的正确姿势yolo detect val会输出 mAP50、mAP50-95、precision、recall。但道路病害任务里mAP 高不代表能用。必须补两个动作第一按类别看指标。val命令加plotsTrue会生成混淆矩阵。如果裂缝的 recall 只有 0.4说明漏检严重得回去查标注或提分辨率。第二看实际推理图。指标是数字推理图是真相。随机抽 20 张验证集图片跑推理肉眼过一遍。我见过 mAP 0.8 但坑槽全标偏的模型指标好看没用。yolo detect val \ modelruns/road/exp1/weights/best.pt \ datadata.yaml \ imgsz1024 \ plotsTrueplotsTrue会在输出目录生成 PR 曲线、混淆矩阵、验证集预测图。重点看混淆矩阵里裂缝和龟裂有没有混——这两类在低分辨率下确实难分如果混得厉害考虑合并类别或提分辨率。4.2 推理脚本与置信度阈值选择推理时conf阈值直接决定漏检和误检的平衡。默认 0.25道路病害建议调到 0.3 到 0.4。太低会把路面纹理误检成裂缝太高会漏掉浅裂缝。from ultralytics import YOLO import cv2 model YOLO(runs/road/exp1/weights/best.pt) results model.predict( sourcetest_images/, conf0.35, iou0.5, imgsz1024, saveTrue, projectinference, nameroad_test ) # 统计每类检出数量 for r in results: boxes r.boxes for cls_id in boxes.cls.unique(): count (boxes.cls cls_id).sum().item() print(f类别 {int(cls_id)}: {count} 个)conf0.35是我在道路病害上常用的起点iou0.5控制 NMS 合并阈值。如果同一处裂缝被重复框降 iou 到 0.4如果相邻病害被合并提 iou 到 0.6。imgsz必须和训练时一致否则精度掉得莫名其妙。4.3 导出 ONNX 与部署前检查训练完的.pt权重部署到生产环境前通常先导出 ONNX。YOLOv8 导出命令很简单但导出后必须验证输出一致性。yolo export modelruns/road/exp1/weights/best.pt formatonnx imgsz1024 opset12opset12兼容性最好别用太新的。导出后拿同一张图分别跑 pt 和 onnx对比框坐标误差超过 1 像素就说明导出有问题。常见原因是动态轴设置不对加dynamicTrue试试。5. 道路病害检测避坑记录五个血泪教训5.1 验证集指标虚高实际推理一塌糊涂现象训练日志 mAP50 到 0.85但拿新图推理几乎全漏。原因数据集按随机帧划分相邻帧高度相似验证集和训练集泄漏。解决按视频段或采集批次划分确保验证集来自不同路段。划分完再跑一次验证指标会掉 10 到 20 个点那才是真实水平。5.2 裂缝检测框断裂成多段现象一条连续裂缝被模型框成三四个小框。原因mosaic 增强把裂缝截断模型学到的是局部片段。另外 NMS 的 iou 阈值太低也会导致相邻框不合并。解决close_mosaic设大一点20 到 30让最后阶段看完整图。推理时 iou 提到 0.6或者后处理做框合并。5.3 训练到一半显存爆炸现象前 30 轮正常第 31 轮突然 OOM。原因YOLOv8 的close_mosaic生效后输入尺寸从 mosaic 的小图变成原图显存占用突增。解决训练时imgsz按原图尺寸设别依赖 mosaic 省显存。或者把close_mosaic设成 0全程开 mosaic但精度会受影响。5.4 类别 ID 错位导致全盘皆输现象模型把所有坑槽都标成裂缝混淆矩阵惨不忍睹。原因data.yaml的names顺序和标注转换时的CLASS_MAP不一致。解决转换脚本和 yaml 文件用同一个配置源别手写两遍。转换完立刻可视化抽查确认类别颜色对得上。5.5 推理速度远低于预期现象训练时 GPU 利用率 90%推理时只有 20%单张图要 200ms。原因推理时没开半精度或者图像预处理在 CPU 上成了瓶颈。解决推理加halfTrue图像读取用cv2.imread后直接传 numpy 数组别走 PIL 绕一圈。如果还慢检查是不是每张图都重新加载了模型。6. 把 YOLOv8 道路病害模型压到 10ms 以内一个具体技巧模型训练完只是开始真正落地时推理速度往往比精度更致命。我做过一个车载巡检项目要求单帧处理不超过 10ms原始 yolov8s 在 1024 尺寸下要 25ms差了一倍多。最后靠三招压到 9ms这里把最有效的一招展开讲。核心思路是「降分辨率 提 batch 半精度」的组合拳但顺序不能乱。先固定 batch1 测基线然后逐步加优化每步记录延迟找到瓶颈再动手。import time import torch from ultralytics import YOLO model YOLO(runs/road/exp1/weights/best.pt) model.to(cuda) # 基线fp32, imgsz1024, batch1 dummy torch.randn(1, 3, 1024, 1024).cuda() for _ in range(10): model.predict(dummy, verboseFalse) torch.cuda.synchronize() t0 time.time() for _ in range(100): model.predict(dummy, verboseFalse) torch.cuda.synchronize() print(fp32 1024:, (time.time() - t0) / 100 * 1000, ms) # 优化1半精度 model.model.half() torch.cuda.synchronize() t0 time.time() for _ in range(100): model.predict(dummy.half(), verboseFalse) torch.cuda.synchronize() print(fp16 1024:, (time.time() - t0) / 100 * 1000, ms) # 优化2降分辨率到 640 dummy640 torch.randn(1, 3, 640, 640).cuda().half() torch.cuda.synchronize() t0 time.time() for _ in range(100): model.predict(dummy640, verboseFalse) torch.cuda.synchronize() print(fp16 640:, (time.time() - t0) / 100 * 1000, ms)实测下来fp16 能省 30% 左右1024 降到 640 再省 50%。但 640 下小裂缝召回率会掉所以最终方案是「640 推理 裂缝区域二次裁剪精检」先用 640 快速筛出疑似区域再把疑似区域裁出来放大到 1024 跑一次。整体延迟 9ms召回率只掉 2 个点。这个技巧的关键在于别指望一个模型解决所有尺度。道路病害的尺度跨度太大分级处理比单模型硬扛划算得多。另外半精度推理时注意如果模型里有自定义算子fp16 可能溢出先在小批量上验证输出是否正常。最后说个习惯每次改完推理参数我都会拿同一批 50 张图跑一遍记录延迟和检出数量画成散点图。延迟降了但检出数量暴跌说明参数调过头了。这个习惯帮我省过好几次线上事故。希望帮到你。本文还有配套的精品资源点击获取
简介本资源面向计算机视觉学习者与智能交通方向开发者提供一套基于Yolov8的道路病害目标检测完整项目覆盖横向裂缝、纵向裂缝、块状裂缝、龟裂、坑槽及多种修补类病害的识别任务适合课程大作业、毕业设计或工程原型验证。压缩包共549个文件约1.76MB以345个md文档、124个py脚本、26个yaml与22个yml配置为主辅以xml、txt、sh及少量前端与部署文件涵盖数据预处理、模型训练、权重保存与加载等完整流程。已有618人学习下载。项目内含源码、模型、数据集说明与项目说明读者可据此复现从数据集准备、参数配置、训练调优到mAP、召回率、精确率评估的全过程并理解如何将模型部署到无人机巡检、车载摄像头等实时场景为智能交通与智慧城市应用提供可迁移的技术参考。1. 道路病害检测从 YOLOv8 开始一份大作业源码到底能跑出什么道路病害目标检测这件事真正做过的人都知道难点从来不在模型本身而在数据。坑槽、裂缝、龟裂、沉陷这几类病害在真实道路影像里尺度差异极大——一条细裂缝可能只占十几个像素一块坑槽却能横跨半个画面。YOLOv8 作为单阶段检测器在这个任务上算是目前性价比最高的选择训练快、部署轻、对小目标也有一定容忍度。这份大作业打包了源码、模型、数据集说明和项目说明本质上是一套「从零跑通道路病害检测」的最小闭环。适合谁课程设计要交东西的学生、想入门工业缺陷检测的工程师、以及手里已经有一批道路巡检图但不知道怎么下手的人。下面我按自己复现这类项目的顺序把每一步拆开讲。2. 道路病害数据集怎么准备从原始标注到 YOLO 格式2.1 先搞清楚你的数据长什么样道路病害数据集通常来自车载巡检设备或手机拍摄分辨率从 1920×1080 到 4000×3000 都有。拿到手第一件事不是急着训练而是统计三件事类别分布、标注格式、图像尺寸分布。类别分布决定你要不要做重采样标注格式决定你要不要写转换脚本图像尺寸分布决定你 imgsz 设多大。常见做法是先用一段脚本把数据集扫一遍输出每个类别的框数量和图像尺寸直方图。这一步花十分钟能省掉后面几小时的无效训练。import os import cv2 from collections import Counter # 假设标注是 YOLO txt 格式每行: class_id cx cy w h label_dir datasets/road_damage/labels/train img_dir datasets/road_damage/images/train cls_counter Counter() size_counter Counter() for txt in os.listdir(label_dir): if not txt.endswith(.txt): continue with open(os.path.join(label_dir, txt)) as f: for line in f: cls_id int(line.split()[0]) cls_counter[cls_id] 1 # 对应图像尺寸 img_path os.path.join(img_dir, txt.replace(.txt, .jpg)) if os.path.exists(img_path): h, w cv2.imread(img_path).shape[:2] size_counter[(w, h)] 1 print(类别分布:, cls_counter) print(尺寸分布 Top5:, size_counter.most_common(5))这段脚本的逻辑很直白遍历标签文件统计类别再读对应图像拿尺寸。参数上唯一要注意的是img_dir和label_dir的对应关系——YOLO 要求图像和标签同名不同后缀如果你的数据集不是这个结构先重命名对齐。类别分布如果严重倾斜比如裂缝占 80%沉陷只占 2%后面训练时要么加类别权重要么对少样本类做增强。2.2 标注格式转换的四个边界坑很多道路病害数据集原始标注是 VOC XML 或 LabelMe JSON转成 YOLO txt 时最容易翻车的地方有四个第一坐标归一化。YOLO 的 cx cy w h 都是相对图像宽高的比例不是绝对像素。转换时除以宽高就完事但要注意浮点精度保留六位小数足够。第二越界框。标注时手抖画到图像外面是常事转换后 cx 可能小于 0 或大于 1。这种框直接裁到边界别留着否则训练时数据增强会报错。第三空标签文件。有些图像没有病害对应 txt 是空的。YOLOv8 默认会跳过空标签但如果你开了rect训练模式空文件可能导致 batch 组装异常。建议要么删掉空图要么在配置里显式处理。第四类别 ID 从 0 开始。VOC 的类别名转 YOLO 时一定要建一个固定的 name-to-id 映射表别用set去重后随机排序否则每次转换 ID 都在变模型根本学不到东西。import xml.etree.ElementTree as ET import os # 固定类别映射顺序一旦定下就不要改 CLASS_MAP {crack: 0, pothole: 1, alligator: 2, subsidence: 3} def voc_to_yolo(xml_path, img_w, img_h, out_path): tree ET.parse(xml_path) root tree.getroot() lines [] for obj in root.findall(object): name obj.find(name).text.strip() if name not in CLASS_MAP: continue cls_id CLASS_MAP[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) # 裁剪越界坐标 x1, x2 max(0, x1), min(img_w, x2) y1, y2 max(0, y1), min(img_h, y2) if x2 x1 or y2 y1: continue cx (x1 x2) / 2 / img_w cy (y1 y2) / 2 / img_h w (x2 - x1) / img_w h (y2 - y1) / img_h lines.append(f{cls_id} {cx:.6f} {cy:.6f} {w:.6f} {h:.6f}) with open(out_path, w) as f: f.write(\n.join(lines))这段代码的关键在CLASS_MAP和越界裁剪。CLASS_MAP必须手动维护别偷懒用自动排序。裁剪那两行是后悔药不加的话后面增强阶段会随机崩。转换完记得抽查几张可视化确认框位置没偏。2.3 数据集划分与 data.yaml 配置YOLOv8 要求一个data.yaml描述数据集路径和类别。典型结构是 train/val/test 三个子集比例 7:2:1 或 8:1:1。道路病害数据如果来自连续视频帧千万别随机划分——相邻帧几乎一样随机划分会导致验证集泄漏指标虚高。正确做法是按视频段或采集日期划分。# data.yaml path: /home/user/road_damage train: images/train val: images/val test: images/test names: 0: crack 1: pothole 2: alligator 3: subsidencepath写绝对路径最稳相对路径在不同工作目录下容易找不到。names的顺序必须和转换时的CLASS_MAP完全一致错一个就是灾难。3. YOLOv8 训练道路病害模型参数怎么设、曲线怎么看3.1 环境配置与最小训练命令环境这块YOLOv8 依赖 ultralytics 包Python 3.8 以上都行。如果你用 GPU先确认 CUDA 和 PyTorch 版本匹配。GTX 1660 Ti 这种 6G 显存的卡跑 yolov8n 或 yolov8s 没问题yolov8m 就要降 batch 了。pip install ultralytics yolo detect train \ datadata.yaml \ modelyolov8n.pt \ epochs100 \ imgsz640 \ batch16 \ device0 \ projectruns/road \ nameexp1这是最小可跑命令。modelyolov8n.pt会用预训练权重道路病害数据量不大时迁移学习比从头训快得多。imgsz640是默认值如果你的裂缝特别细可以提到 1024但显存占用会翻倍。batch16在 6G 卡上跑 640 尺寸刚好爆显存就降到 8。3.2 道路病害场景下的关键参数默认参数能跑但想跑好这几个参数必须调imgsz裂缝类病害建议 1024坑槽类 640 够用。判断依据是标注框的最小边长——如果大量框小于 20 像素就得提分辨率。batch显存允许范围内越大越好但别超过 32否则学习率要跟着调。小数据集用 8 或 16 更稳。lr0初始学习率默认 0.01。道路病害数据如果只有几千张降到 0.001 更稳避免早期震荡。patience早停轮数默认 50。小数据集设 20 就够省时间。close_mosaic最后多少轮关闭 mosaic 增强默认 10。道路病害的裂缝在 mosaic 拼接后容易断裂建议设 20让模型最后阶段看到真实分布。yolo detect train \ datadata.yaml \ modelyolov8s.pt \ epochs200 \ imgsz1024 \ batch8 \ lr00.001 \ patience20 \ close_mosaic20 \ device0这套参数是我在类似任务上试出来的折中值。imgsz 提到 1024 后小裂缝召回率能涨 5 到 8 个点但训练时间翻倍。如果时间紧640 也能出活只是细裂缝漏检多。3.3 损失曲线怎么读翻车信号识别训练启动后runs/road/exp1/下会生成results.csv和损失曲线图。重点看三条线box_loss、cls_loss、dfl_loss。正常情况三条都单调下降最后趋于平缓。翻车信号一box_loss 下降但 cls_loss 不降。说明模型能定位但分不清类别通常是类别不平衡或标注噪声。解决方法是检查标注或者给少样本类加权。翻车信号二val_loss 先降后升。典型过拟合加数据增强或减模型容量。道路病害数据增强推荐用mosaic0.5、flipud0.5、hsv_h0.015别开太大旋转裂缝方向是重要特征。翻车信号三mAP 卡在 0.3 以下不动。先别怀疑模型去可视化验证集预测结果。八成是标注格式错了或者类别 ID 对不上。import pandas as pd import matplotlib.pyplot as plt df pd.read_csv(runs/road/exp1/results.csv) df.columns df.columns.str.strip() fig, axes plt.subplots(1, 2, figsize(12, 4)) axes[0].plot(df[epoch], df[train/box_loss], labelbox) axes[0].plot(df[epoch], df[train/cls_loss], labelcls) axes[0].plot(df[epoch], df[train/dfl_loss], labeldfl) axes[0].legend() axes[0].set_title(Train Loss) axes[1].plot(df[epoch], df[metrics/mAP50(B)], labelmAP50) axes[1].plot(df[epoch], df[metrics/mAP50-95(B)], labelmAP50-95) axes[1].legend() axes[1].set_title(Val mAP) plt.savefig(curve.png)这段脚本把训练日志画成两张图。注意列名可能有空格strip()先处理掉。mAP50 到 0.6 以上算可用0.75 以上算不错。道路病害因为标注主观性强别追求 0.9不现实。4. 模型验证与推理mAP 之外还要看什么4.1 验证集评估的正确姿势yolo detect val会输出 mAP50、mAP50-95、precision、recall。但道路病害任务里mAP 高不代表能用。必须补两个动作第一按类别看指标。val命令加plotsTrue会生成混淆矩阵。如果裂缝的 recall 只有 0.4说明漏检严重得回去查标注或提分辨率。第二看实际推理图。指标是数字推理图是真相。随机抽 20 张验证集图片跑推理肉眼过一遍。我见过 mAP 0.8 但坑槽全标偏的模型指标好看没用。yolo detect val \ modelruns/road/exp1/weights/best.pt \ datadata.yaml \ imgsz1024 \ plotsTrueplotsTrue会在输出目录生成 PR 曲线、混淆矩阵、验证集预测图。重点看混淆矩阵里裂缝和龟裂有没有混——这两类在低分辨率下确实难分如果混得厉害考虑合并类别或提分辨率。4.2 推理脚本与置信度阈值选择推理时conf阈值直接决定漏检和误检的平衡。默认 0.25道路病害建议调到 0.3 到 0.4。太低会把路面纹理误检成裂缝太高会漏掉浅裂缝。from ultralytics import YOLO import cv2 model YOLO(runs/road/exp1/weights/best.pt) results model.predict( sourcetest_images/, conf0.35, iou0.5, imgsz1024, saveTrue, projectinference, nameroad_test ) # 统计每类检出数量 for r in results: boxes r.boxes for cls_id in boxes.cls.unique(): count (boxes.cls cls_id).sum().item() print(f类别 {int(cls_id)}: {count} 个)conf0.35是我在道路病害上常用的起点iou0.5控制 NMS 合并阈值。如果同一处裂缝被重复框降 iou 到 0.4如果相邻病害被合并提 iou 到 0.6。imgsz必须和训练时一致否则精度掉得莫名其妙。4.3 导出 ONNX 与部署前检查训练完的.pt权重部署到生产环境前通常先导出 ONNX。YOLOv8 导出命令很简单但导出后必须验证输出一致性。yolo export modelruns/road/exp1/weights/best.pt formatonnx imgsz1024 opset12opset12兼容性最好别用太新的。导出后拿同一张图分别跑 pt 和 onnx对比框坐标误差超过 1 像素就说明导出有问题。常见原因是动态轴设置不对加dynamicTrue试试。5. 道路病害检测避坑记录五个血泪教训5.1 验证集指标虚高实际推理一塌糊涂现象训练日志 mAP50 到 0.85但拿新图推理几乎全漏。原因数据集按随机帧划分相邻帧高度相似验证集和训练集泄漏。解决按视频段或采集批次划分确保验证集来自不同路段。划分完再跑一次验证指标会掉 10 到 20 个点那才是真实水平。5.2 裂缝检测框断裂成多段现象一条连续裂缝被模型框成三四个小框。原因mosaic 增强把裂缝截断模型学到的是局部片段。另外 NMS 的 iou 阈值太低也会导致相邻框不合并。解决close_mosaic设大一点20 到 30让最后阶段看完整图。推理时 iou 提到 0.6或者后处理做框合并。5.3 训练到一半显存爆炸现象前 30 轮正常第 31 轮突然 OOM。原因YOLOv8 的close_mosaic生效后输入尺寸从 mosaic 的小图变成原图显存占用突增。解决训练时imgsz按原图尺寸设别依赖 mosaic 省显存。或者把close_mosaic设成 0全程开 mosaic但精度会受影响。5.4 类别 ID 错位导致全盘皆输现象模型把所有坑槽都标成裂缝混淆矩阵惨不忍睹。原因data.yaml的names顺序和标注转换时的CLASS_MAP不一致。解决转换脚本和 yaml 文件用同一个配置源别手写两遍。转换完立刻可视化抽查确认类别颜色对得上。5.5 推理速度远低于预期现象训练时 GPU 利用率 90%推理时只有 20%单张图要 200ms。原因推理时没开半精度或者图像预处理在 CPU 上成了瓶颈。解决推理加halfTrue图像读取用cv2.imread后直接传 numpy 数组别走 PIL 绕一圈。如果还慢检查是不是每张图都重新加载了模型。6. 把 YOLOv8 道路病害模型压到 10ms 以内一个具体技巧模型训练完只是开始真正落地时推理速度往往比精度更致命。我做过一个车载巡检项目要求单帧处理不超过 10ms原始 yolov8s 在 1024 尺寸下要 25ms差了一倍多。最后靠三招压到 9ms这里把最有效的一招展开讲。核心思路是「降分辨率 提 batch 半精度」的组合拳但顺序不能乱。先固定 batch1 测基线然后逐步加优化每步记录延迟找到瓶颈再动手。import time import torch from ultralytics import YOLO model YOLO(runs/road/exp1/weights/best.pt) model.to(cuda) # 基线fp32, imgsz1024, batch1 dummy torch.randn(1, 3, 1024, 1024).cuda() for _ in range(10): model.predict(dummy, verboseFalse) torch.cuda.synchronize() t0 time.time() for _ in range(100): model.predict(dummy, verboseFalse) torch.cuda.synchronize() print(fp32 1024:, (time.time() - t0) / 100 * 1000, ms) # 优化1半精度 model.model.half() torch.cuda.synchronize() t0 time.time() for _ in range(100): model.predict(dummy.half(), verboseFalse) torch.cuda.synchronize() print(fp16 1024:, (time.time() - t0) / 100 * 1000, ms) # 优化2降分辨率到 640 dummy640 torch.randn(1, 3, 640, 640).cuda().half() torch.cuda.synchronize() t0 time.time() for _ in range(100): model.predict(dummy640, verboseFalse) torch.cuda.synchronize() print(fp16 640:, (time.time() - t0) / 100 * 1000, ms)实测下来fp16 能省 30% 左右1024 降到 640 再省 50%。但 640 下小裂缝召回率会掉所以最终方案是「640 推理 裂缝区域二次裁剪精检」先用 640 快速筛出疑似区域再把疑似区域裁出来放大到 1024 跑一次。整体延迟 9ms召回率只掉 2 个点。这个技巧的关键在于别指望一个模型解决所有尺度。道路病害的尺度跨度太大分级处理比单模型硬扛划算得多。另外半精度推理时注意如果模型里有自定义算子fp16 可能溢出先在小批量上验证输出是否正常。最后说个习惯每次改完推理参数我都会拿同一批 50 张图跑一遍记录延迟和检出数量画成散点图。延迟降了但检出数量暴跌说明参数调过头了。这个习惯帮我省过好几次线上事故。希望帮到你。本文还有配套的精品资源点击获取
