YOLOv8+PyTorch花卉图像识别毕设完整实践指南

YOLOv8 做花卉图像识别是目前毕设里比较稳妥的一种组合目标检测框架负责把图里的花定位出来并标出类别PyTorch 负责承接模型训练和推理。这类项目真正值钱的不是把 Demo 跑通而是把数据集整理、标注格式、训练参数、结果评估这一整条链路都摸清楚。适合正在选毕设题目的本科生也适合刚接触深度学习的同学。下面按我实际踩过的顺序从环境到训练再到排查完整拆一遍。1. 先说清楚这个毕设项目到底做的是什么1.1 YOLOv8 是检测框架还是分类框架先别搞混任务类型很多人一提 YOLOv8就默认它只能画框检测。实际上 YOLOv8 这个系列包含检测、分类、分割、姿态估计四类模型调用起来分别是yolov8n.pt、yolov8n-cls.pt、yolov8n-seg.pt、yolov8n-pose.pt。花卉图像识别也存在两种理解图像分类整张图里只有一种花判断它是玫瑰、向日葵还是郁金香。这种任务用-cls系列模型数据集按类别文件夹组织就行。目标检测一张图里有多种花或者花在图片角落里需要把每一朵花用矩形框定位出来并给出类别。这种任务用标准的yolov8n.pt检测模型需要边界框标签。毕设里大多数情况做的是检测因为检测任务的展示效果更好能画出框、给出置信度写论文时也更容易做实验对比。标题里写的是“基于 YOLOv8PyTorch 框架的花卉图像识别”如果没有特殊说明建议直接按目标检测来做。1.2 花卉图像识别和通用目标检测有什么区别通用目标检测的数据集往往包含人、车、猫、狗、杯子这类类别差异很大的对象模型比较容易区分。花卉不一样很多品种之间差异很小比如玫瑰和月季花瓣颜色相近叶子形状相似同一品种在不同光线、不同角度下看起来又像完全不同的东西。这就带来三个实际问题标注难度高花和背景的边界不清晰花瓣重叠严重标注框稍微偏一点训练结果就会受影响。类别不均衡常见数据集里玫瑰、向日葵样本多某种野花样本少模型容易把样本少的类别漏掉或误判。对图像增强敏感裁剪、旋转、色彩抖动对花卉识别的影响比其他任务更大因为颜色和纹理本身就是重要特征。所以做这个项目时不要花太多时间纠结“模型选哪个”先把数据集质量搞扎实。数据不行换再大的模型也救不回来。1.3 完整项目链路包含哪些模块一个能写进毕业设计的项目至少应该包含下面这些环节数据集收集与整理图片标注画框 标类别标注格式转换成 YOLO 需要的 txt 格式编写数据集配置文件data.yaml选择预训练模型并启动训练查看训练曲线和验证指标用训练好的权重做单图和批量推理导出模型做成简单界面或接口很多同学只做到第 6 步就以为完了结果答辩时老师问“你模型部署用过吗”“批量图片怎么处理”答不上来。所以后面会把推理和导出也展开讲。2. 环境准备把 PyTorch 和 YOLOv8 跑起来2.1 确认硬件条件别急着上最大模型先看自己电脑是什么情况。训练 YOLOv8 并不是非要有高端显卡不可但硬件直接决定你能用哪个规格的模型和多大的图片尺寸。我建议先按这个标准判断硬件条件推荐模型建议图片尺寸训练批次无 GPU纯 CPUyolov8n320 或 4162~44GB 显存yolov8n 或 yolov8s416 或 6404~86GB 显存如 GTX 1660 Tiyolov8s6408~168GB 以上显存yolov8m64016~3216GB 以上显存yolov8l / yolov8x640 或 1280按显存调GTX 1660 Ti 跑 YOLOv8simgsz640batch8是能正常训练的只是每个 epoch 时间会长一些。如果你的机器只有 CPU也能跑但训练时间会非常久。这种情况下不要做完整训练先用 1 个 epoch 验证流程再决定要不要换云 GPU 或减少数据量。2.2 安装 PyTorch 和 Ultralytics 的推荐顺序环境安装最常见的坑是 PyTorch 没装好导致torch.cuda.is_available()返回 False后面所有训练都跑在 CPU 上。安装顺序我建议固定为先装对应版本的 Python建议 3.8 到 3.11。再到 PyTorch 官网生成安装命令按自己的 CUDA 版本选择。安装完成后再安装 ultralytics 包。最后检查 PyTorch 是否能识别 GPU。为什么要按这个顺序因为 ultralytics 依赖 torch 和 torchvision如果先装了 ultralyticspip 会自动帮你装一份默认的 torch很大概率是 CPU 版本。等你训练的时候才发现用的是 CPU再卸载重装很容易把依赖搞乱。PyTorch 安装好之后建议先执行一次python -c import torch; print(torch.__version__); print(CUDA:, torch.cuda.is_available())如果第二行输出CUDA: True说明 PyTorch 已经识别到显卡可以继续。如果输出CUDA: False先别急着跑 YOLO回去检查驱动、CUDA 和 PyTorch 版本是否匹配。AMD 显卡也能跑 PyTorch但需要走 ROCm 分支Windows 下兼容性不如 NVIDIA 稳定如果对这套不熟刚开始用 CPU 跑也没问题。2.3 安装 ultralytics 并做最小推理测试确认 PyTorch 正常之后再装 ultralyticspip install ultralytics安装完成后用一段最简单的 Python 代码验证环境from ultralytics import YOLO # 第一次运行会下载 yolov8n 的预训练权重 model YOLO(yolov8n.pt) # 换成你自己电脑上的一张测试图片路径 results model(test.jpg) print(len(results[0].boxes))如果代码能输出检测框数量说明整个环境链路已经通了。这一步很关键它能帮你把“环境问题”和“项目问题”隔离开。环境没跑通之前不要开始整理数据集和训练否则后面报错了无法判断是环境的问题还是数据的问题。3. 数据集组织这是毕设最容易翻车的地方3.1 数据集应该长什么样图片和标签一一对应YOLOv8 训练时要求数据目录有固定的结构。最常用的组织方式是这样的flower_dataset/ ├── images/ │ ├── train/ │ │ ├── flower_001.jpg │ │ ├── flower_002.jpg │ │ └── ... │ └── val/ │ ├── flower_010.jpg │ └── ... ├── labels/ │ ├── train/ │ │ ├── flower_001.txt │ │ ├── flower_002.txt │ │ └── ... │ └── val/ │ ├── flower_010.txt │ └── ... └── data.yaml这里的硬性要求是每张图片对应一个同名 txt 文件。images/train/flower_001.jpg必须对应labels/train/flower_001.txt。文件名不一致、txt 放错子目录、train 和 val 的图片标签对不上这三种情况是最常见的报错来源。如果使用的是公开花卉数据集比如 Oxford 102 Flowers 或者其他整理好的数据集通常需要写一个转换脚本把原本的数据格式转成上述结构。如果你拿到的“完整数据集”已经整理成了这个格式那就可以直接跳到写data.yaml的环节。3.2 标注格式转换从普通图像标注到 YOLO txt 格式YOLO 的 txt 标签格式每一行代表一个目标内容为类别ID x_center y_center width height五个值都是归一化后的数值范围在 0 到 1 之间。例如0 0.5234 0.3211 0.1234 0.2345这表示第一个目标类别是 0边界框中心点在图片宽度 52.34% 的位置高度 32.11% 的位置框宽占图片宽度的 12.34%框高占图片高度的 23.45%。如果你用 LabelImg、LabelMe、CVAT 等工具标注要导出 YOLO 格式。Roboflow 也支持在线标注并一键导出缺点是需要上传图片数据量太大时体验一般。这里有一个常见误区类别 ID 必须从 0 开始连续编号。如果数据集有 5 类花类别编号就是 0、1、2、3、4不能跳号。data.yaml里的names列表顺序也必须和标签里的类别 ID 一一对应否则训练出来的模型类别全是乱的。3.3 划分训练集和验证集划分数据集建议用脚本做不要手动拖文件夹。因为手动划分很容易出现训练集和验证集分布不一致的问题。比如训练集里全是晴天拍的玫瑰验证集里全是阴天拍的玫瑰那么训练时模型只学到了一部分特征验证时表现会突然变差。一般按 8:2 或 9:1 划分。划分时固定随机种子保证每次结果可以复现。同时最好按类别分布分层抽样也就是每一类花在训练集和验证集里的比例尽量一致。import os import random import shutil random.seed(42) image_root flower_dataset/images/all label_root flower_dataset/labels/all train_img_dir flower_dataset/images/train val_img_dir flower_dataset/images/val train_lab_dir flower_dataset/labels/train val_lab_dir flower_dataset/labels/val os.makedirs(train_img_dir, exist_okTrue) os.makedirs(val_img_dir, exist_okTrue) os.makedirs(train_lab_dir, exist_okTrue) os.makedirs(val_lab_dir, exist_okTrue) files [f for f in os.listdir(image_root) if f.endswith(.jpg)] random.shuffle(files) split int(len(files) * 0.8) for f in files[:split]: shutil.move(os.path.join(image_root, f), train_img_dir) shutil.move(os.path.join(label_root, f.replace(.jpg, .txt)), train_lab_dir) for f in files[split:]: shutil.move(os.path.join(image_root, f), val_img_dir) shutil.move(os.path.join(label_root, f.replace(.jpg, .txt)), val_lab_dir)这只是示例脚本实际使用时把路径改成自己的目录结构。如果你的数据集规模比较小比如每类只有几十张建议先做数据增强再做划分否则验证集的代表性会不足。3.4 编写 data.yaml训练之前必须要有一个数据配置文件。下面是一个典型示例# 数据集根目录可以写绝对路径也可以写成相对路径 path: D:/flower_dataset train: images/train val: images/val names: 0: rose 1: sunflower 2: tulip 3: daisy 4: dandelion注意几点names的类别数量必须和标注里的实际类别数一致。类别名不要用中文尽量用英文或拼音避免编码问题。train和val的路径是相对于path的路径不是绝对路径。写完data.yaml后可以用一段小脚本验证一下标签和图片是否一一对应并统计每类样本数量。这一步能避免训练到一半才发现 label 文件是空的。4. 训练模型从单轮开始再跑完整训练4.1 先用小参数跑通流程不要一上来就开完整训练我的习惯是先把整条流程用最小配置跑一遍。所谓最小配置就是只训练 1 个 epoch用很小的 batch图片尺寸也降到 320。这样做的原因是如果数据路径错了、配置文件写错了、标签格式不对小配置下能快速暴露问题几秒钟就能看到报错。等流程跑通之后再开完整训练。yolo train modelyolov8n.pt dataflower_dataset/data.yaml epochs1 imgsz320 batch2 device0跑完这个命令你要确认三件事终端里是否出现正常的训练进度。是否自动生成了runs/detect/train目录。目录里有没有weights/best.pt和weights/last.pt。这三个条件都满足说明数据链路和训练链路是通的可以放心进入正式训练。4.2 完整训练的关键参数怎么调正式训练时核心参数不外乎下面这些参数含义建议值判断标准model预训练权重yolov8n.pt或yolov8s.pt显存 4GB 以内用 n6GB 用 s8GB 以上可考虑 mepochs训练轮数100~200看验证集指标是否收敛imgsz输入图片尺寸640显存不够时降到 416 或 320batch每次迭代的样本数-1自动检测或用 8/16显存报错时减小device训练设备0或cpu用0表示第一块 GPUpatience早停轮数20连续 20 轮验证集指标不提升就停止lr0初始学习率默认即可曲线震荡剧烈时调小cache是否缓存图片True小数据集可以开内存不足时关闭不要一上来就把参数拉满。imgsz1280、batch64看起来好像更厉害但显存不够会直接 OOM而且对于花卉识别这种任务640 已经足够更大的尺寸只会增加训练时间不一定会提升多少精度。训练命令示例yolo train modelyolov8s.pt dataflower_dataset/data.yaml epochs150 imgsz640 batch8 device0 patience20 projectflower_train namerun1加project和name是为了让不同实验的结果分开存放不会互相覆盖。后面做对比实验时这个习惯会省很多时间。4.3 训练过程怎么观察损失曲线怎么看训练过程中的日志会输出box_loss、cls_loss、dfl_loss、precision、recall、mAP50、mAP50-95等指标。很多同学只看最终 mAP忽略了训练过程。实际上过程信息能帮你判断很多问题。训练结束后在runs/detect/train目录下会生成results.png里面包含训练损失曲线和验证指标曲线。如果你想自己画损失函数曲线图可以直接读取训练时自动保存的results.csv用 pandas 和 matplotlib 绘制。看曲线时要记住几个判断标准训练损失下降验证损失也在下降说明模型还在正常学习。训练损失下降验证损失不再下降甚至上升说明开始过拟合了。损失曲线出现剧烈震荡可能学习率偏高或者 batch 太小。mAP50 在早期不涨是正常的不要焦虑。一般地训练到后面验证集的 mAP50 趋向平稳就可以停下来了。早停机制的作用就体现在这里patience20表示连续 20 个 epoch 验证集指标没有变好会自动保存最优权重并停止训练。4.4 如何做增量训练不重新训练全部有时候项目中途会补充数据或者需要在新数据集上继续训练。这种情况属于增量训练YOLOv8 是支持的。第一种情况在同一批类别上继续训练。直接用之前训练好的best.pt作为model继续跑。yolo train modelflower_train/run1/weights/best.pt dataflower_dataset/data.yaml epochs50 imgsz640 batch8 device0第二种情况如果模型之前训练过但你现在换了全新的数据集类别完全不同那么训练时要用预训练权重而不是旧权重。因为旧权重的输出头是为旧类别设计的直接当作起点效果不一定好。第三种情况如果训练中途断了想接着跑使用yolo train modelflower_train/run1/weights/last.pt dataflower_dataset/data.yaml epochs150 resumeTrue增量训练时要注意如果新增了类别模型输出头会重建前几个 epoch 的损失会有明显震荡这是正常现象。不要因为前几轮指标暂时不好就立刻放弃给模型多留一些训练轮数。5. 模型验证和推理让模型真正“识别”花5.1 验证集上评估mAP、精确率、召回率怎么看训练结束后第一个动作是跑验证集yolo val modelflower_train/run1/weights/best.pt dataflower_dataset/data.yaml这条命令会输出验证集上的整体指标包括每种类别的 Precision、Recall 和 mAP。重点看两个数mAP50IoU 阈值为 0.5 时的平均精度。这个指标更容易涨适合初步判断模型是否学到了东西。mAP50-95IoU 阈值从 0.5 到 0.95 的平均精度。这个指标更严格也更接近实际部署场景下的检测效果。如果 mAP50 很高但 mAP50-95 很低说明模型只能画出大致位置框的定位精度不够。常见原因是标注框本身不够精确或者图片尺寸偏小导致细粒度特征丢失。训练生成的runs/detect/train目录下还有confusion_matrix.png、PR_curve.png、F1_curve.png。这些图在论文里很有用尤其是 PR 曲线和混淆矩阵能直观展示模型在各类别上的表现差异。5.2 单张图片推理和批量推理推理是最能体现项目完整度的环节。单张图片推理yolo predict modelflower_train/run1/weights/best.pt sourcetest_images/rose.jpg saveTrue批量处理图片时source可以直接指向文件夹yolo predict modelflower_train/run1/weights/best.pt sourcetest_images saveTrue这条命令会自动处理文件夹里所有图片并把标注结果图片保存到runs/detect/predict目录。实际项目里如果要做量化分析建议用 Python 接口遍历结果并统计识别正确率、每张耗时等数据from ultralytics import YOLO model YOLO(flower_train/run1/weights/best.pt) results model(test_images/, saveTrue) for r in results: boxes r.boxes print(r.path, len(boxes), boxes.cls.tolist(), boxes.conf.tolist())批量推理时注意一个坑如果你把几千张图片一次性传入程序会把所有图片的结果都保存在内存里内存不够时会被杀进程。建议用streamTrue按生成器方式逐张处理results model(test_images/, saveTrue, streamTrue) for r in results: # 这里每处理一张就返回一个结果 pass5.3 导出部署ONNX 和半精度推理训练好的 PyTorch 模型如果要给其他人用或者放进一个界面里单纯用.pt文件并不是最优方案。.pt文件需要依赖 PyTorch 环境别人拿去也要装一整套环境才能跑。导出成 ONNX 是一个常见做法yolo export modelflower_train/run1/weights/best.pt formatonnx导出之后可以用 ONNX Runtime 加载不依赖 PyTorch 也能推理。如果你的电脑有 NVIDIA 显卡还可以导出 TensorRT engineyolo export modelflower_train/run1/weights/best.pt formatengine推理速度会明显提升但生成的 engine 文件只对当前显卡有效换一台机器需要重新导出。这里顺便说说 FP32、FP16、BF16、TF32。默认训练保存的权重是 FP32精度完整但体积大、计算慢。如果只做展示和部署可以转成 FP16 半精度推理显存占用下降速度提升精度损失通常很小。BF16 的数值范围和 FP32 更接近适合某些训练场景但对硬件有要求。TF32 是 NVIDIA Ampere 及以上架构里的计算模式一般由框架自动使用。毕设阶段不需要深入研究所有格式记住一点PyTorch 默认 FP32 稳妥部署到 GPU 上追求速度时可以试 FP16。6. 常见问题排查训练卡住、指标为 0、显存不足怎么办6.1 启动阶段就报错先查环境再查数据训练命令刚敲下去就报错通常不是模型问题。按这个顺序排查module not found类的错误先确认 ultralytics、torch、torchvision 装没装版本是否匹配。torch.cuda.is_available()返回 False确认 CUDA 驱动、PyTorch 版本、显卡驱动三者是否匹配。File not found错误查看data.yaml里的路径特别注意反斜杠和正斜杠Windows 下不要用单个反斜杠写路径。No labels found错误检查标签目录名字有没有拼错应该叫labels不叫label检查图片和 txt 文件名是否完全一致。中文路径问题数据集路径和文件路径尽量不要包含中文和空格很多字符处理问题都是中文路径引起的。6.2 训练过程中显存不足显存不足是最常见的问题报错关键词一般是OutOfMemory或CUDA out of memory。处理方法只有几个调小batch从 16 调到 8再不行调到 4。调小imgsz从 640 调到 416 或 320。换更小的模型从yolov8s换到yolov8n。关闭cache参数因为缓存图片会额外占用内存。降低数据加载线程数用workers2或workers4看任务管理器里的占用情况。不要想着用梯度累计或者强行降低精度来硬撑毕设阶段没有必要为了一个参数把环境搞得特别复杂。6.3 训练能跑但 mAP 一直为 0这种情况大概率是数据问题。排查顺序是随机打开几个 txt 标签文件看类别 ID 是否在data.yaml的names范围内。查看标注框的坐标值是否在 0 到 1 之间。有些标注工具导出的是像素坐标需要先归一化。确认训练集和验证集图片里真的有目标。有时候验证集里全是空图mAP 自然为 0。确认标注文件没有被误清空。可以用脚本统计一下每个 txt 文件的内容行数空文件多的数据集要重新检查。6.4 训练结果不理想先改数据再改参数如果训练完成但实际推理效果很差不要急着调学习率。建议按这个顺序找原因先看是不是过拟合。训练损失很低但验证损失高说明模型把训练数据背下来了应该增加数据量、增加数据增强或者减小模型规模。再看是不是类别不均衡。如果某类花样本特别少模型大概率把这类漏检。解决方法是补充样本或者对少数类做过采样。最后再调整训练参数。常见做法是降低学习率、增加训练轮数、调整置信度阈值。图片识别效果不好时先从输入图片角度检查图片太小、分辨率太低、花在图像中占比过小都会导致检测不到。把训练图片和推理图片的分辨率对齐很多怪问题就消失了。7. 把项目做成完整毕设界面、对比实验和论文结构7.1 增加一个简单界面让模型从命令行变成可演示工具如果毕设答辩时只展示命令行输出效果会弱很多。建议做一个简单的 Web 界面让人上传一张图片后台调用训练好的模型返回标注结果图片和识别类别。用 Gradio 是一个很轻量的方案几十行代码就能完成import gradio as gr from ultralytics import YOLO model YOLO(flower_train/run1/weights/best.pt) def detect(image): result model(image) result[0].save(output.jpg) return output.jpg gr.Interface(fndetect, inputsimage, outputsimage).launch()这段代码只是示例实际使用时可以调整返回格式把类别名和置信度也显示出来。Gradio 默认会启动一个本地 Web 服务浏览器打开就可见答辩时演示非常直观。7.2 增加对比实验不同模型规模和不同训练方式毕设论文里实验对比是不可或缺的一部分。不要只训练一个模型就收工。建议做下面两类对比不同模型规格对比分别训练yolov8n、yolov8s、yolov8m对比 mAP、训练时间、模型大小、推理速度。不同训练策略对比从零训练和用预训练权重微调验证集指标的差异。最后可以用一张表格总结例如模型图片尺寸mAP50mAP50-95单张推理耗时模型体积yolov8n6400.8820.63112ms6.2MByolov8s6400.9140.68220ms22.5MByolov8m6400.9260.70135ms49.7MB上面表格里的数据是示例不代表真实结果。你自己训练后把实际数据填进去就行。有这样一个表格论文的实验章节写起来会轻松很多。7.3 增加数据增强和复杂场景测试花卉识别在自然场景里的难点主要是遮挡、密集花丛、光线变化、背景干扰。可以在测试时加入一些额外样本倾斜拍摄、俯拍、仰拍的花卉图片有遮挡的花卉图片比如部分藏在叶子后面不同天气和光照条件下的图片背景复杂的图片比如花在草丛、街道、花盆里如果模型在这些场景下表现一般可以在训练时打开数据增强。YOLOv8 的增强参数里hsv_h、hsv_s、hsv_v控制颜色扰动fliplr、flipud控制翻转mosaic控制多图拼接。训练时默认开启一部分增强一般不需要手动调太多直接使用默认值即可。7.4 论文结构怎么安排这类基于 YOLOv8 的花卉识别毕设论文章节可以按这样的逻辑安排绪论选题背景、国内外研究现状、研究内容。相关技术深度学习基础、卷积神经网络、YOLOv8 原理。数据集构建数据来源、预处理、标注、数据集划分。模型训练与优化环境、预训练权重、训练参数、实验结果。系统设计与实现界面、推理流程、整体功能展示。实验结果分析对比实验、典型错误分析、优化方向。总结与展望。写实验章节的时候不要只贴图表要把“为什么这样设置参数”“如果换一种方案会怎样”写清楚。答辩老师最喜欢问的就是这些细节。踩过几次之后我发现这类项目真正难的地方不是 YOLOv8 本身而是数据集太乱、环境没配好、批量任务一开始就开满参数。建议先把单条训练流程跑稳确认图片、标签、日志都正常再开完整训练。等模型训练好之后再做界面、对比实验和论文图表。按这个顺序走1 小时跑通 Demo 是完全可行的后面的优化才有意义。