基于深度学习的茶叶等级检测系统（含UI界面、yolov8、Python代码、数据集）-CSDN博客

本文链接：https://blog.csdn.net/qq_28949847/article/details/141609254

在这里插入图片描述

项目介绍

项目中所用到的算法模型和数据集等信息如下：

算法模型：
yolov8、yolov8 + SE注意力机制，两种模型都已训练好，可直接使用。

数据集：
网上下载的数据集，格式都已转好，可直接使用。

以上是本套代码算法的简单说明，添加注意力机制可作为bishe、作业的创新点 。如果要是需要其他的检测模型，请私信。

在这里插入图片描述

项目简介

本文将详细介绍如何以官方yolov8为主干，实现对茶叶等级的检测识别，且利用PyQt5设计了简约的系统UI界面。在界面中，您可以选择自己的视频文件、图片文件进行检测。此外，您还可以更换自己训练的主干模型，进行自己数据的检测。

该系统界面优美，检测精度高，功能强大。它具备多目标实时检测，同时可以自由选择感兴趣的检测目标。

本博文提供了完整的Python程序代码和使用教程，适合新入门的朋友参考。您可以在文末的下载链接中获取完整的代码资源文件。以下是本博文的目录：

功能展示：

部分核心功能如下：

功能1： 支持单张图片识别
功能2： 支持遍历文件夹识别
功能3： 支持识别视频文件
功能4： 支持摄像头识别
功能5： 支持结果文件导出（xls格式）
功能6： 支持切换检测到的目标查看

更多的其他功能可以通过下方视频演示查看。

基于深度学习的茶叶等级检测识别系统（yolov8）

🌟 一、环境安装

本项目提供所有需要的环境安装包（python、pycharm、cuda、torch等），提供的各个库之间的版本都是匹配好的可以直接按照视频讲解或者文档进行安装。

这个指导文档是我经过三年多的精心总结（是我辅导的上千人安装环境遇到的问题总结），它汇集了详细的安装步骤以及在环境安装过程中常见的问题和解决方案。这份文档不仅全面，而且会持续更新，可以持续关注此文档。

不管你是GPU版，还是CPU版，文档中都提供了详细的安装步骤，包含各种安装细节，真心不能再详细了😂😂😂😂😂😂。。。。

环境安装指导文档链接：https://aax3oiawuo.feishu.cn/wiki/XRXFwJH9AiIV2Ok02eMca5SUnNe?from=from_copylink
在这里插入图片描述
而且B站提供有和文档对应的视频，可以将视频和文档结合起来观看，更容易理解。环境安装对应的视频可以在我B站上自己对应的项目里找到，示例截图如下：

在这里插入图片描述

🌟 二、数据集介绍

提供全面、结构化的数据集，它不仅包含了丰富的类别，而且已经细致地划分为训练集、验证集和测试集，以满足不同阶段的模型训练需求。而且数据集的格式，可直接支持YOLO训练，无需额外的格式转换工作。

数据样式如下：

在这里插入图片描述

🌟 三、yolov8相关介绍

YOLOv8 是一个 SOTA 模型，它建立在以前 YOLO 版本的成功基础上，并引入了新的功能和改进，以进一步提升性能和灵活性。具体创新包括一个新的骨干网络、一个新的 Ancher-Free 检测头和一个新的损失函数，可以在从 CPU 到 GPU 的各种硬件平台上运行。

不过 ultralytics 并没有直接将开源库命名为 YOLOv8，而是直接使用 ultralytics 这个词，原因是 ultralytics 将这个库定位为算法框架，而非某一个特定算法，一个主要特点是可扩展性。其希望这个库不仅仅能够用于 YOLO 系列模型，而是能够支持非 YOLO 模型以及分类分割姿态估计等各类任务。
总而言之，ultralytics 开源库的两个主要优点是：

融合众多当前 SOTA 技术于一体
未来将支持其他 YOLO 系列以及 YOLO 之外的更多算法

在这里插入图片描述

网络结构如下：
在这里插入图片描述

🌟 四、模型训练步骤

提供封装好的训练脚本，如下图，更加详细的的操作步骤可以参考我的飞书在线文档：https://aax3oiawuo.feishu.cn/wiki/RqCQw0yBMie3WMkpOwOcilUCnsg ，强烈建议直接看文档去训练模型，文档是实时更新的，有任何的新问题，我都会实时的更新上去。另外B站也会提供视频。

使用pycharm打开代码，找到train.py打开，示例截图如下：
修改 model_yaml 的值，根据自己的实际情况修改，想要训练 yolov8s模型就修改为 model_yaml = yaml_yolov8s，训练添加SE注意力机制的模型就修改为 model_yaml = yaml_yolov8_SE
修改data_path 数据集路径，我这里默认指定的是traindata.yaml 文件，如果训练我提供的数据，可以不用改

修改 model.train()中的参数，按照自己的需求和电脑硬件的情况更改

# 文档中对参数有详细的说明
model.train(data=data_path,             # 数据集
            imgsz=640,                  # 训练图片大小
            epochs=200,                 # 训练的轮次
            batch=2,                    # 训练batch
            workers=0,                  # 加载数据线程数
            device='0',                 # 使用显卡
            optimizer='SGD',            # 优化器
            project='runs/train',       # 模型保存路径
            name=name,                  # 模型保存命名
            )

修改traindata.yaml文件，打开 traindata.yaml 文件，如下所示：

在这里，只需修改 path 的值，其他的都不用改动（仔细看上面的黄色字体），我提供的数据集默认都是到 yolo 文件夹，设置到 yolo 这一级即可，修改完后，返回 train.py 中，执行train.py。
打开 train.py ，右键执行。
出现如下类似的界面代表开始训练了
训练完后的模型保存在runs/train文件夹下

🌟 五、模型评估步骤

打开val.py文件，如下图所示：
修改 model_pt 的值，是自己想要评估的模型路径
修改 data_path ，根据自己的实际情况修改，具体如何修改，查看上方模型训练中的修改步骤

修改 model.val()中的参数，按照自己的需求和电脑硬件的情况更改

model.val(data=data_path,           # 数据集路径
          imgsz=300,                # 图片大小，要和训练时一样
          batch=4,                  # batch
          workers=0,                # 加载数据线程数
          conf=0.001,               # 设置检测的最小置信度阈值。置信度低于此阈值的检测将被丢弃。
          iou=0.6,                  # 设置非最大抑制 (NMS) 的交叉重叠 (IoU) 阈值。有助于减少重复检测。
          device='0',               # 使用显卡
          project='runs/val',       # 保存路径
          name='exp',               # 保存命名
          )