🏋️模型训练
通过深度学习网络可以从大量数据中学习数据特征，从而应用于各种检测，分割任务。想要使用深度学习解决具体问题，必须了解如何通过已知数据进行网络模型的训练。因此我们首先讲解了如何训练深度学习神经网络。本节的内容如下：
在本地进行模型训练
在云端进行模型训练
制作自己的训练集
在本地进行模型训练
本小节介绍如何在个人PC上进行模型训练，您需要拥有以下硬件：
一台带有英伟达显卡的笔记本或台式机，安装Windows系统(如果您的电脑并不含有英伟达显卡，那么我们建议您使用云端进行模型训练。)
请安装Anaconda软件用来管理您的python环境。Anaconda有Windows,Max,Linux三个版本可以选择，请根据您的系统选择对应的版本。
安装VScode作为您的代码编辑软件。
使用anaconda创建我们训练模型要使用的环境：
下面打开Anaconda Propmt
输入以下命令，进入工作命令，激活环境，打开vscode（注意，将工作目录和环境名称替换成你自己的相应的名称）。
在Terminal里输入以下命令：
完成后输入
pip uninstall torch torchvision
卸载后安装GPU适用的torch版本
安装前请查看您显卡所安装的cuda驱动，安装对应cuda版本的pytorch及torchvision。例如，我的cuda版本高于12.1,所以我安装
这时你已经完成了模型训练所需的所有环境配置，下面就可以愉快的训练模型了😊。
训练模型需要执行以下命令：
其中，model为你所需要训练的模型路径，目前的Grove Vision V2只支持yolov8n.pt的部署，所以请选择yolov8n.pt。
data为模型训练所需训练集的路径，在这里我们使用一个很小的数据集作为测试，在本节的后面会讲解如何创建训练集进行训练。另外我们实现了coco数据集的自定义部分标签训练，您可以按照自己的喜好以及用途选择特定标签进行训练，实现更灵活的推理功能。
imgsz为模型输入接受的图像大小，这里必须为192
epochs为训练的次数，一般来说，训练次数越多，模型与数据拟合程度越好。
训练结束后，您可以在命令行显示的模型保存的位置找到您训练后的模型（例如我的模型保存在runs\detect\train3）
在云端进行模型训练
如果您的电脑不带有英伟达显卡，那么很可惜您无法在本地进行模型训练，不过您可以在云端进行模型训练，两种训练方法得到的模型效果是相同的。
我们使用ultralytics进行云端模型训练。以下是 Ultralytics HUB 的链接，打开链接，你就可以看到它的介绍和使用教程。
要使用 Ultralytics HUB，你需要一个 github 账户或谷歌账户或苹果账户。
ultralytics hub支持训练用户自己的数据集，如何制作并上传数据集请查看之后章节。
设置高级模型配置（Advanced Model Configuration），如果以后使用 Google Colab 进行模型训练，建议将 Epochs 改为 30，因为虽然 Google Colab 是免费的，但使用有时间限制，30 次训练约为 2 小时，正好在时间限制范围内。如果要在 Grove Vision V2 模块上部署模型，还需要将图像大小改为 192，这更符合我们的实际应用场景。
此外，使用 Google Colab 还需要 Google 邮箱。
制作自己的训练集
如果您想要对开源数据集之外的标签进行检测，那么您需要自己制作数据集。数据集应该包括图像和对应标签。
下面是yolov8指定的数据集格式
我们使用Make Sense在线数据标注工具来完成训练集制作过程：
有关如何使用Make Sense，可以参考：
当您完成了数据集标签的标注，您需要将图像和标签数据合并起来，构成一个完整的数据集。
数据集需要满足以下格式：
您需要把Make Sense生成的标签数据拷贝到相应的位置。比如您标记了用于训练的标签数据，那么训练所需的图像应该位于images/train，而标签数据应该位于labels/train。另外您需要编写一个dataset.yaml文件。该文件需要放在dataset目录下，文件内容为：
（！注意：dataset.yaml命名要与文件夹命名相一致，否则数据集上传到ultralytics时会遇到问题）
names需要和Make Sense标注时的标签相对应。
如果您在本地进行训练，建议您将train，val，test的路径修改为绝对路径，而在云端训练则不需要。
到此步骤您已经可以使用制作好的数据集进行模型训练了，只需要将
中的“data=coco128.yaml”路径改为您制作的数据集中dataset.yaml即可。
如果您在云端进行训练，那么您需要将数据集上传到ultralytics hub上，上传过程非常简单：
这样Ultralytics HUB中就存有了您的数据集，您此时便可以使用您自己的数据集进行模型训练。
Last updated
