YOLOv8官方教程：如何训练模型

一、简介

本文介绍YOLOv8如何在单GPU、多GPU和Apple设备上训练数据。

二、YOLOv8的优点

• 效率：充分榨干硬件性能，无论是单GPU还是多GPU。源码可以好好学一学。
• 多功能性：可以在 COCO、VOC 和 ImageNet数据集上训练模型，包括自定义数据集。
• 用户友好：可以用命令行快速启动训练，也可以用几行python代码启动。
• 超参数灵活性：各种可定制的超参数可微调模型性能。

train模式特点

以下是 YOLOv8 Train 模式的一些显着特点：

• 自动下载数据集：首次使用时会自动下载 COCO、VOC 和 ImageNet 等标准数据集。
• 多 GPU 支持：能够加快训练速度。
• 超参数配置：通过 YAML 配置文件或 CLI 参数修改超参数的选项。
• 可视化和监控：可视化训练指标和训练过程。

YOLOv8 数据集（例如 COCO、VOC、ImageNet 等）会在首次使用时自动下载，
即yolo train data=coco.yaml

三、训练

在 COCO128 数据集上，训练输入图像分辨率为 640X640， 训练 YOLOv8n 100 个epoch。有关训练参数的完整解释，请参阅下面的参数部分。

3.1 单 GPU 和 CPU 训练示例

有GPU代码自动使用GPU，没有就自动切换为CPU训练。python代码：
【重要的事情说三遍】：

• yolov8n.yaml yolov8n.pt coco128.yaml自动下载
• yolov8n.yaml yolov8n.pt coco128.yaml自动下载
• yolov8n.yaml yolov8n.pt coco128.yaml自动下载

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.yaml')  # build a new model from YAML
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
# 或者这么写
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  # build from YAML and transfer weights

# Train the model
results = model.train(data='coco128.yaml', epochs=100, imgsz=640)

或者在命令行执行（以下三条指令，选择执行一条就行）：

# Build a new model from YAML and start training from scratch
yolo detect train data=coco128.yaml model=yolov8n.yaml epochs=100 imgsz=640

# Start training from a pretrained *.pt model
yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640

# Build a new model from YAML, transfer pretrained weights to it and start training
yolo detect train data=coco128.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 imgsz=640

3.2 多GPU训练

如果你有多个显卡，可以使用参数指定训练显卡。默认使用第0块显卡。

多GPU训练示例python代码：

from ultralytics import YOLO
# Load a model
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)
# Train the model with 2 GPUs
results = model.train(data='coco128.yaml', epochs=100, imgsz=640, device=[0, 1])

或者使用命令行

# Start training from a pretrained *.pt model using GPUs 0 and 1
yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640 device=0,1

3.3 Apple M1 和 M2 MPS 训练

Apple自带处理器也可以用来加速训练。小数据集还差不多，数据集一大，卡得一匹。
python代码：

from ultralytics import YOLO

# Load a model
model = YOLO('yolov8n.pt')  # load a pretrained model (recommended for training)

# Train the model with 2 GPUs
results = model.train(data='coco128.yaml', epochs=100, imgsz=640, device='mps')

或者使用下面命令行：

# Start training from a pretrained *.pt model using GPUs 0 and 1
yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640 device=mps

3.4 恢复中断的训练

下面是如何使用 Python 和通过命令行恢复中断的训练的示例：
python代码：

from ultralytics import YOLO
# Load a model
model = YOLO('path/to/last.pt')  # load a partially trained model
# Resume training
results = model.train(resume=True)

或者在命令行执行：

# Resume an interrupted training
yolo train resume model=path/to/last.pt

通过设置resume=True，恢复训练。其中，train函数将将加载“path/to/last.pt模型”文件，模型文件中包含模型参数和上一次正常训练的各种超参数。如果resume设置为False，那就重新训练呗。还有就是，从last.pt文件中恢复训练，必须满足条件：last.pt至少已经训练了一个epoch。

四、其他参数

上文给了很多命令行参数，这里给出其他参数以及相关解释。(下表翻译看看就好)

五、训练Log

在训练 YOLOv8 模型时，需要记录下训练过程中，模型的loss，各项性能指标等。Ultralytics 的 YOLO 提供对三种类型记录器的支持 - Comet、ClearML 和 TensorBoard。只需要安装这些库，训练过程总将自动调用。本文强力推荐tensorboard。

5.1 TensorBoard

TensorBoard本来是TensorFlow 的可视化工具。后来pytorch也集成了，用户还是非常多的。别看tensorflow做得跟shit一样，但是tensorboard做的还是很不错的。它可以可视化训练loss、各种验证指标、超参数等等。
执行下面明令：

tensorboard --logdir ultralytics/runs  # replace with 'runs' directory

bash窗口会打印例如：http://localhost:6006/ 这样的信息，复制到浏览器，打开后效果如下图：

5.2 Comet

Comet是一个允许数据科学家和开发人员跟踪、比较、解释和优化实验和模型的平台。它提供实时指标、代码差异和超参数跟踪等功能。

使用commet：

# pip install comet_ml
import comet_ml
comet_ml.init()

请记住在其网站上登录您的 Comet 帐户并获取您的 API 密钥。您需要将其添加到环境变量或脚本中以记录实验。

5.3 ClearML

ClearML是一个开源平台，可自动跟踪实验并帮助高效共享资源。它旨在帮助团队更有效地管理、执行和重现他们的机器学习工作。

要使用 ClearML：

# pip install clearml
import clearml

clearml.browser_login()

运行此脚本后，您需要在浏览器上登录您的 ClearML 帐户并验证您的会话。