简介面向玉米叶病害检测的开发者与学习者这套资源提供了YOLOv8训练权重、PyQt可视化界面与一千五百张标注数据集覆盖常见锈病、灰斑病、叶枯病及健康叶片四类可直接用于模型推理或桌面端交互演示。压缩包总计两千个文件其中一千七百个txt标签保存检测框信息一百七十一个Python脚本负责训练、推理与界面逻辑五十六个yaml描述数据集路径及类别另有预训练pt权重、ui界面文件及说明文档整包约一百八十二兆。数据集已按训练集、验证集、测试集划分并配好配置文件YOLOv5、YOLOv7、YOLOv8、YOLOv9均可直接训练标签为文本格式迁移学习门槛较低。目前已有四百三十六人学习下载适合需要完整工程化方案的初学者或竞赛选手既能基于权重快速验证效果也可借助PyQt界面直观查看检测结果省去从数据整理到环境配置的重复工作。1. 玉米叶病害检测这套组合1500 张图、一组权重、一个 PyQt 桌面工具1500 张玉米叶病害图像这个体量在公开数据集里排不上号但配上 YOLOv8 官方预训练权重做微调再用 PyQt 包一层鼠标点击就能用的桌面界面就是一套在实验站、乡镇农技站里能跑起来的落地方案。这类组合在农业视觉项目里很常见数据不是海量检测目标是叶片上的病斑或整片发病叶最终交付物不是训练脚本而是一个不依赖命令行、打开就能加载权重选图的软件。这篇文章从数据集准备、权重训练写到界面封装和交付前验证适合刚拿 YOLOv8 训练自己数据集的新手也适合已经跑通过检测、想把模型变成工具的工程师。2. 玉米病害数据集准备从 1500 张原始照片到干净的训练目录2.1 先定义检测目标病斑、发病叶片还是整株分等级拿到 1500 张图第一件事不是急着标框而是想清楚检测对象。同一个“玉米叶病害”可以拆成三种粒度把单张叶片上的一块病斑框出来把一张发病的叶片整体框出来或者按整株玉米的发病程度分等级。这三种做法训练难度完全不同界面展示效果也差很多。我一般倾向于先做“病斑 叶片”两层目标也就是把明显的锈病孢子堆、灰斑病长条斑单独框同时给整片发病严重的叶子一个叶片级框。原因是病斑框能训练出模型对小目标的敏感度叶片框容错率高用户在界面上看结果时更容易信服。如果一上来就做八类病害平均每类分不到两百张很多类别的框数量会跌破几十个YOLOv8 再怎么调参都很难出稳定权重。还有个容易被忽略的动作标注前先统计每类图片数量和框数量。1500 张图听着不少但假如其中 1300 张是锈病、只有 200 张是灰斑病那你实际是在用一个严重偏斜的数据集。训练出来的权重会看着总 mAP 还行单独看灰斑病那一类可能只有 0.2 出头。这个统计在标注刚开始时就要做别等标完 1500 张再回头补类。2.2 采集与清洗哪些照片必须丢玉米叶病害照片大多来自手机、田间相机和无人机采集质量参差不齐。清洗阶段我一般丢三类图模糊到看不清叶脉纹理的、叶片占比过小或完全被土壤背景淹没的、以及同一片叶在同一角度连续拍摄的连拍图。模糊图会逼着模型去学边缘噪声连拍图则会让训练集里隐性地重复同一张图最终导致验证集指标虚高。清洗的落地做法不复杂。先按文件名和拍摄时间排一遍把间隔小于一秒的连拍删到只剩一张再用一个简单的脚本统计图像分辨率和亮度分布把分辨率明显低于其余样本的图单独挪出来人工看。标注时如果发现某张图里目标小到只有十几个像素我会直接删掉而不是勉强打框因为 YOLOv8 在 640 输入下对这些小目标学习效率很低还容易把背景学进特征里。注意清洗后的图片编号最好重排一次避免文件名里带着原始拍摄批次信息防止后续切分数据集时无意识地把同一批照片全部放进验证集。2.3 数据增强把小数据集用出增量同时避免泄漏1500 张图对目标检测来说偏少但 YOLOv8 训练时自带的 mosaic、翻转、HSV 扰动已经在起作用所以不一定需要额外做大量离线增强。离线增强适合两类情况一是某类病害图片实在少需要复制几份平衡类别二是希望模型对特定的田间光照变化更鲁棒比如阴天和正午强光。做离线增强时有个非常容易翻车的点增强出来的图片不能进验证集。很多人把翻转、旋转后的图混进同一个文件夹再随机切分结果训练集和验证集里出现同一张原图的变体验证 mAP 会虚高 5 到 10 个点部署到新照片上立刻原形毕露。增强图只放训练集验证集和测试集必须用原始照片。下面是一个最小可用的水平翻转增强脚本同时会改写 YOLO 标注文件里的 x 坐标import cv2 img cv2.imread(leaf_001.jpg) h, w img.shape[:2] # 读取 YOLO 格式标注每行 class x_center y_center box_w box_h lines open(leaf_001.txt, r).readlines() new_lines [] for line in lines: parts line.strip().split() cls, xc, yc, bw, bh parts[0], float(parts[1]), float(parts[2]), float(parts[3]), float(parts[4]) new_xc 1.0 - xc # 水平翻转后中心 x 变为 1-x new_lines.append(f{cls} {new_xc:.6f} {yc:.6f} {bw:.6f} {bh:.6f}\n) cv2.imwrite(leaf_001_flip.jpg, cv2.flip(img, 1)) with open(leaf_001_flip.txt, w) as f: f.writelines(new_lines)这段代码的核心是new_xc 1.0 - xc。YOLO 的坐标是归一化到 0 到 1 的水平翻转时图像左右镜像目标中心 x 会从原来的值变成镜像后的位置翻转 y 不变。宽度高度也都不用变。如果你用的是垂直翻转或旋转 90 度坐标变换规则要重新推导不要套用这个公式。实操中我会把这份脚本整理成一个遍历目录的版本只对选定的少数类图片执行并且输出到单独的aug_train目录避免污染原始数据。2.4 标注格式与目录切分YOLO txt 与 8:1:1 脚本标注工具的选择直接影响后续工作量。玉米叶病害这种带弧度的目标用 LabelImg 或 X-AnyLabeling 都行关键是输出格式统一成 YOLO txt。如果你用的工具默认输出 VOC 的 xml记得转换后再进训练否则 YOLOv8 的 data.yaml 读不到标注。每张图片对应一个同名 txt里面一行一个框格式是类别编号加四个归一化小数。切分数据集时我坚持按“图片名哈希”划分而不是简单地把前 80% 给训练集。按目录顺序切分容易把相同拍摄条件下的图集中到训练集或验证集导致验证结果失真。下面这个脚本按文件名的 md5 值做 8:1:1 切分并生成 data.yamlimport os, random, hashlib, glob image_paths sorted(glob.glob(images/*.jpg)) random.seed(42) buckets {train: [], val: [], test: []} for img_path in image_paths: digest hashlib.md5(os.path.basename(img_path).encode()).hexdigest() r int(digest, 16) % 10 # 用哈希前几位转整数取 0-9 if r 8: buckets[train].append(img_path) elif r 9: buckets[val].append(img_path) else: buckets[test].append(img_path) for split, files in buckets.items(): os.makedirs(fdatasets/corn/{split}/images, exist_okTrue) os.makedirs(fdatasets/corn/{split}/labels, exist_okTrue) for img_path in files: base os.path.splitext(os.path.basename(img_path))[0] label_path flabels/{base}.txt os.rename(img_path, fdatasets/corn/{split}/images/{os.path.basename(img_path)}) os.rename(label_path, fdatasets/corn/{split}/labels/{base}.txt)这段脚本的关键是hashlib.md5那行。用文件名的哈希值取余数能保证每次重新执行时划分结果稳定不会因为列表顺序变化导致每次训练集都不一样。random.seed(42)保留是为了需要随机打乱的场景实际哈希划分里它不是主导因素。8:1:1 的比例对 1500 张图意味着训练集约 1200 张、验证集约 150 张、测试集约 150 张验证集和测试集虽然不大但作为没见过的照片做最终评估已经够用。切分完成后data.yaml 里要注意train、val写到一级目录names列表的顺序必须和标注时的类别编号一一对应。我见过最多次的翻车就是 names 顺序写错导致界面里显示的名字和实际框对不上。3. 训练玉米叶病害权重预训练选择、参数与损失曲线3.1 基座模型选型yolov8n 还是 yolov8sYOLOv8 官方预训练权重按体积分为 n、s、m、l、x 几档玉米叶病害这类任务我推荐从 s 起步。n 模型虽小但叶片病斑这种纹理差异较小的目标n 的特征提取能力容易不够m 及以上的模型在 1500 张数据上更容易过拟合训练时间也成倍增加。如果你手里只有 GTX 1660 Ti 这类 6GB 显存s 配合 batch 16 是比较稳妥的组合。模型参数量级6GB 显存可跑 batch推理延迟适用场景yolov8n最小较大最快快速初筛、边缘设备yolov8s较小16 左右快小数据集微调首选yolov8m中等8 左右中等数据量 3000 时考虑yolov8l较大4 左右较慢数据充足、追求精度还有一个容易被忽略的点modelyolov8s.pt这一行写的是官方在 COCO 上预训练好的权重不是从头训练。YOLOv8 加载这个权重后会保留 backbone 对边缘、纹理、颜色的基础特征玉米叶片和 COCO 里的植物类目有一定相关性所以微调收敛很快。如果误用了yolov8s.yaml而不是.pt就会进入从随机权重开始训练的状态1500 张图很难训出可用精度。3.2 关键训练参数epochs、batch、imgsz 与早停判断用预训练权重微调时我一般从 200 个 epoch 起步patience 设 50。1500 张图属于中等偏小的数据集几十个 epoch 后验证集 mAP 就会进入平台期没必要硬跑满 200 轮但 epoch 太少又可能没等到 mosaic 增强阶段结束就停了。batch 的选择看显存s 模型在 6GB 显存上 batch 16 没问题显存不够就降到 8同时把 lr0 从 0.001 调到 0.0007 左右因为小 batch 的梯度噪声更大学习率要跟着保守一点。imgsz 我习惯固定 640。玉米叶病害图上病斑尺寸不大imgsz 640 已经能覆盖大多数框调到 1280 理论上对小病斑更友好但训练时间和显存占用会明显上升对于 1500 张图的小项目性价比不高。还有两个容易被忽略的参数close_mosaic和cos_lr。close_mosaic设为 10让最后 10 个 epoch 关闭 mosaic 增强能避免 mosaic 拼接出的虚假边缘干扰模型最后的精调cos_lr打开后学习率衰减更平滑对小数据集更友好。3.3 训练命令与损失曲线怎么确认权重真的在收敛训练命令我一般写成下面这样所有输出保留在独立的 runs 目录下方便后面翻看yolo detect train \ datadatasets/corn/data.yaml \ modelyolov8s.pt \ epochs200 \ imgsz640 \ batch16 \ lr00.001 \ patience50 \ close_mosaic10 \ cos_lrTrue \ projectruns \ namecorn_yolov8s \ device0这里的device0指定第一块 GPU没有 GPU 就写devicecpu但训练速度会慢非常多。projectruns和namecorn_yolov8s是为了让每次训练结果落在独立的文件夹里不会相互覆盖。训练开始后不用一直盯着终端YOLOv8 会把每个 epoch 的指标写进runs/corn_yolov8s/results.csv这才是判断训练状态的核心文件。等训练结束或早停触发后画损失曲线图是最直观的验收动作。直接用 pandas 读 results.csv把 train/loss 和 val/loss 两条曲线叠在一起看import pandas as pd import matplotlib.pyplot as plt df pd.read_csv(runs/corn_yolov8s/results.csv) df.columns [c.strip() for c in df.columns] epochs df[epoch] train_loss df[train/box_loss] df[train/cls_loss] df[train/dfl_loss] val_loss df[val/box_loss] df[val/cls_loss] df[val/dfl_loss] plt.plot(epochs, train_loss, labeltrain loss) plt.plot(epochs, val_loss, labelval loss) plt.xlabel(epoch) plt.ylabel(loss) plt.legend() plt.grid(True) plt.savefig(corn_loss_curve.png, dpi150)代码里train/box_loss、train/cls_loss、train/dfl_loss是 YOLOv8 输出的三个损失分量加总后看整体趋势。判断权重好坏不能只看训练损失降得多快关键是 val loss 有没有在某个 epoch 后掉头向上。如果 val loss 持续上升说明正在过拟合应该以最低点之前的权重为准而不是用最后一个 epoch 的权重。画这张图的价值在于你能清楚地看到它是在第 60 个 epoch 还是第 120 个 epoch 进入平台期从而决定下一轮训练要不要减 epoch 或增强数据。3.4 权重控制与选型best.pt、last.pt 与权重备份训练结束后runs 目录下会生成best.pt和last.pt两个权重文件。best 是验证集指标最好的那一轮权重last 是最后一个 epoch 的权重。界面部署时我会毫不犹豫地选 best但如果早停触发得早best 和 last 差别很大说明训练后期过拟合严重需要回去检查数据增强或学习率。best.pt 同时包含了模型结构、类别名和训练配置加载时不需要额外再传 yaml 文件。提到权重控制我要多说一句不少人把 best.pt 复制到界面目录后又回头继续训练同一个project/name结果 best.pt 被新训练覆盖界面里加载的权重悄悄变了这就是典型的权重控制失败场景。我习惯在训练一结束就把验证集 mAP 和对应的 best.pt 复制到weights/corn_best_xx.pt文件名里带日期或指标值比如corn_best_0.832.pt。这样界面加载的权重始终是明确的不会因为后续实验被覆盖。如果你后续要调置信度阈值或做更多实验这个备份习惯能省掉很多确认成本。4. 用 PyQt 把权重封装成可点界面的检测工具4.1 界面组成与数据流从权重到 QLabel 上的检测框PyQt 界面的任务不是把 YOLOv8 的 API 再包一层而是让用户打开软件、选择一张田间照片、点一下按钮就能看到带标注框的结果。我常用的界面结构是在 QMainWindow 里放三个区域左侧是按钮和参数面板包括打开图片、摄像头检测、批量检测、置信度滑条和权重选择右侧中间是结果显示 QLabel下方是日志区显示每次检测的图片名、类别和置信度。这个界面背后的数据流是主线程只负责接收按钮事件并把任务丢给后台线程后台线程加载权重、执行模型推理把检测结果通过 Qt 信号传回主线程主线程再把图像画到 QLabel 上。最忌讳的做法是在按钮点击回调里直接调用model.predict()因为推理是阻塞操作处理一张 640 图片在 GPU 上还好在 CPU 上可能要几百毫秒界面会直接卡住。4.2 推理线程点击后界面不卡死的通用模板下面是一个标准的 QThread 推理模板适用于打开图片和摄像头两种模式。核心是把模型加载和预测都放到子线程pyqtSignal把结果送回主线程from PyQt5.QtCore import QThread, pyqtSignal import cv2 class DetectThread(QThread): result_ready pyqtSignal(object, object) def __init__(self, model, parentNone): super().__init__(parent) self.model model # 在进入线程前加载好模型 self.images [] # 待检测图片路径列表 self.running False def set_images(self, paths): self.images paths def run(self): self.running True for path in self.images: if not self.running: break img cv2.imread(path) results self.model.predict(img, conf0.35, imgsz640, verboseFalse) boxes results[0].boxes.xyxy.cpu().numpy() # 像素坐标 classes results[0].boxes.cls.cpu().numpy().astype(int) scores results[0].boxes.conf.cpu().numpy() self.result_ready.emit(img, (boxes, classes, scores)) self.running False这里有个容易踩的坑self.model是在主线程开始前创建并传入的但predict调用发生在子线程里。Ultralytics 的模型对象在推理时支持跨线程调用只要不是同时从两个线程并发推理就行。pyqtSignal(object, object)用来传递 numpy 数组因为 numpy 对象不是 Qt 原生类型直接传会报警告甚至丢数据用 object 类型让 Qt 保持引用即可。conf0.35是推理时的置信度阈值这个值建议在界面上做成可调参数而不是写死在代码里。调用方式是在主窗口里实例化线程并连接信号self.detect_thread DetectThread(self.model) self.detect_thread.result_ready.connect(self.show_result) # 用户点击按钮后 self.detect_thread.set_images([selected_path]) self.detect_thread.start()注意start()之后不要再对线程里的model做其他调用。关闭窗口时要在closeEvent里把running置为 False然后调用wait()否则程序会在退出时报 crashed。4.3 图像坐标换算与画框从 BGR 到 QImage 再到 QPainter子线程把原图传回主线程后主线程要做两件事把 OpenCV 的 BGR 格式转成 QImage再用 QPainter 在图片上画框。转换用下面的代码重点在于copy()def show_result(self, frame, data): boxes, classes, scores data rgb cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) h, w, ch rgb.shape qimg QImage(rgb.data, w, h, ch * w, QImage.Format_RGB888).copy() pixmap QPixmap.fromImage(qimg) painter QPainter(pixmap) painter.setPen(QPen(QColor(#FF3333), 2)) font painter.font() font.setPixelSize(18) painter.setFont(font) for box, cls_id, score in zip(boxes, classes, scores): x1, y1, x2, y2 box.astype(int) painter.drawRect(x1, y1, x2 - x1, y2 - y1) painter.drawText(x1, max(0, y1 - 6), self.model.names[cls_id]) painter.end() # 按 QLabel 大小等比缩放显示 self.result_label.setPixmap(pixmap.scaled( self.result_label.size(), Qt.KeepAspectRatio, Qt.SmoothTransformation))这段代码里最容易被忽略的是QImage(...).copy()。rgb.data指向的是函数内的局部 numpy 数组如果不用copy()QImage 会持有已经释放的内存显示出来的图片会出现花屏或随机色块。检测框的坐标直接使用 YOLOv8 返回的 xyxy 像素坐标因为model.predict()在默认输入尺寸下会把原图缩放后推理输出坐标已经换算回原图尺寸不需要再手动映射。如果你用的是letterbox之后手动前向推理坐标还原就是另一套逻辑了。drawText的 y 坐标用了max(0, y1 - 6)是为了避免目标贴住图片上边缘时文字被画出边界。实际项目里我还会给不同类别分配不同颜色这个可以维护一个类别到颜色的字典避免全部用同一种红。4.4 批量检测与导出把结果写成 CSV 和标注图界面除了单张检测通常还需要批量处理一个文件里的所有图片。批量模式可以直接复用上面的 DetectThread只需要把set_images换成整个目录的图片路径列表。每张图的结果接收后我会同时把检测信息追加到一个二维列表等线程结束后一起写入 CSV。import csv def on_batch_done(self): with open(corn_detect_result.csv, w, newline, encodingutf-8) as f: writer csv.writer(f) writer.writerow([image, class, confidence, x1, y1, x2, y2]) for name, dets in self.batch_results.items(): for det in dets: writer.writerow([name, det[0], f{det[1]:.3f}, *[int(v) for v in det[2]]])CSV 里记录图片名、类别名、置信度和边框坐标方便后续在 Excel 里按病害类别筛选和统计。批量检测时不要在信号回调里写文件因为每张图都触发一次写盘会拖慢整个流程正确的做法是把结果先累积在内存线程跑完后再一次性落盘。如果你需要把检测后的图片也导出来可以在画框后的 pixmap 上调用save()注意保存时用原图分辨率而不是缩放后的 QLabel 尺寸。5. 落地避坑5 条影响精度的真实问题与排查5.1 训练时验证损失掉头向上mAP 反而在涨现象是训练中期开始val loss 曲线持续上升但 mAP 曲线还在缓慢增长很多人看到 mAP 涨就觉得权重没问题。原因在于 YOLOv8 的 mAP 是基于 IoU 和置信度的排序指标即便模型对部分样本过度自信只要排序关系没崩mAP 就可能继续涨而损失函数考量的是分类和定位的精确程度过拟合后损失必然反弹。解决方法是不要只看 mAP盯着 val loss 的最低点来决定权重同时把close_mosaic打开让最后 10 个 epoch 用真实分布微调能显著缓解这种掉头现象。5.2 新地块照片误检率高模型只在训练数据上有效现象是训练集和验证集上 mAP 都在 0.85 以上换一个拍摄角度、光照条件或不同生长阶段的玉米叶立刻出现整片误检。原因多数是训练集太干净所有图片都来自同一块田、同一部手机背景纹理和叶片姿态高度一致模型把背景特征也学进去了。解决这个问题的第一步是清洗时故意保留一部分脏照片比如带泥土背景、带手指遮挡、光线偏暗的图第二步是在验证时专门留出 50 张完全不同来源的照片做盲测。如果盲测 mAP 掉到 0.6 以下最务实的做法是补充新数据而不是继续调参。5.3 PyQt 界面一点按钮就卡死或白屏现象是点击开始检测后窗口失去响应macOS 上还会出现彩色转圈Windows 上是未响应标签。原因是模型推理直接跑在了主线程里predict()的阻塞时间远大于界面刷新周期。解决方法是按 4.2 节的 QThread 方案改造把推理放进子线程这里有个细节QThread被垃圾回收时也会导致程序异常退出所以要像例子那样把线程对象保存为窗口的成员变量。还有一个常见连带错误在子线程里更新 QLabel这同样会引发崩溃。信号只能从子线程发射界面更新必须在主线程的槽函数里完成。5.4 检测框显示位置偏移标在了完全错误的地方现象是单张跑出来的坐标和实际标注位置对不上框明显往左上或右下偏。这个问题的原因多半不在模型而在显示环节如果你拿到results[0].boxes.xyxy后又对图片做了一次缩放、裁剪或者letterbox而没有反向换算坐标画框就会偏。解决办法是在显示链路里统一用原图坐标界面缩放只影响 QLabel 的显示不影响画框前的坐标。如果你确实需要对输入图做 resize 加速推理那么所有坐标都必须按原图宽/输入宽和原图高/输入高两个比例分别映射回来除非 YOLOv8 已经帮你做了这一步。5.5 类别不均衡导致少数类病害几乎漏检现象是训练完看每类 AP数据量大的锈病有 0.9灰斑病只有 0.2但总 mAP 还是被拉得很高表面上看不出来。原因是 1500 张图里灰斑病只占 200 张而且病斑特征和锈病重叠严重。解决的思路不是简单地把灰斑病图片复制三份而是先看每类的框数量如果少于训练集总框数的 10%就把这类图片单独做增强包括对病斑区域的局部裁剪、旋转和亮度扰动。另一种做法是把实在难以区分的类别先合并成其他叶部病害等后续采集到更多数据再拆细。这个取舍虽然牺牲了分类细度但能让界面上的误检率明显下降。6. 交付前的最后一公里置信度调优与权重导出验证界面接上权重只是第一步模型交付前我还会做两个动作调置信度阈值和验证导出后的权重一致性。置信度阈值不要直接沿用训练时的默认值训练时为了计算 mAP 会放宽阈值而实际使用中误检的代价更高。我会从 test 集里抽 100 张真实环境照片分别用 0.25、0.35、0.45 跑一遍统计每张图的误检数量和漏检数量找一个平衡点。玉米叶病害场景里我通常会把最终阈值定在 0.35 到 0.45 之间宁可少检出几个低置信度病斑也不要给用户一堆假框。如果后续打算把模型部署到边缘设备或者 RK3588 这类盒子上需要先导出 ONNX 再转换。导出命令很简单yolo export modelweights/corn_best_0.832.pt formatonnx imgsz640导出后用 onnxruntime 跑同一张图和 .pt 的输出做一致性对比这一步能提前暴露很多算子兼容问题。踩过的坑里最值得记的一条是不要因为 PyQt 界面显示正常的检测结果就跳过 ONNX 验证。界面里用的是 .pt 权重而实际部署跑的是转换后的格式两者输出在坐标和类别上偶尔会有细微偏差。我现在习惯在训练结束后统一把所有权重备份到带指标的文件名导出前再确认一次阈值和类别顺序。这套从 1500 张数据到可用的 YOLOv8 权重、再到 PyQt 界面的链路最大的价值不在某个单点技术而是把数据、模型、界面三个环节打通了。做到的边界是没有高质量新数据模型泛化能力就到头了没有合理的类别定义界面再好看也只是空中楼阁。希望这篇笔记能帮你少走几步弯路顺利把玉米叶病害检测工具跑起来。本文还有配套的精品资源点击获取