yolo task=detect mode=train model=yolov8n.pt       args...
          classify    predict     yolov8n-cls.yaml args...
          segment     val         yolov8n-seg.yaml args...
          export                  yolov8n.pt       format=onnx args...

使用Ultralytics YOLO进行模型训练

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

from ultralytics import YOLO


model = YOLO('yolov8n-pose.yaml')  
model = YOLO('model/yolov8n-pose.pt')  
model = YOLO('yolov8n-pose.yaml').load('yolov8n-pose.pt')  


results = model.train(data='coco8-pose.yaml', epochs=100, imgsz=640)

yolo pose train data=coco8-pose.yaml model=yolov8n-pose.yaml epochs=100 imgsz=640


yolo pose train data=coco8-pose.yaml model=yolov8n-pose.pt epochs=100 imgsz=640


yolo pose train data=coco8-pose.yaml model=yolov8n-pose.yaml pretrained=yolov8n-pose.pt epochs=100 imgsz=640

from ultralytics import YOLO


model = YOLO('yolov8n.yaml')  
model = YOLO('model/yolov8n.pt')  
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  


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

yolo detect train data=coco128.yaml model=yolov8n.yaml epochs=100 imgsz=640


yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640


yolo detect train data=coco128.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 imgsz=640

在COCO128数据集上训练YOLOv8n模型100个时期，图像大小为640。可以使用device
参数指定训练设备。如果没有传递参数，并且有可用的GPU，则将使用GPU device=0，否则将使用device=cpu。有关完整列表的训练参数，请参见下面的参数部分。

设备将自动确定。如果有可用的GPU，那么将使用它，否则将在CPU上开始训练。

from ultralytics import YOLO


model = YOLO('yolov8n.yaml')  
model = YOLO('model/yolov8n.pt')  
model = YOLO('yolov8n.yaml').load('yolov8n.pt')  


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

yolo detect train data=coco128.yaml model=yolov8n.yaml epochs=100 imgsz=640


yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640


yolo detect train data=coco128.yaml model=yolov8n.yaml pretrained=yolov8n.pt epochs=100 imgsz=640

多GPU训练通过在多个GPU上分布训练负载，实现对可用硬件资源的更有效利用。无论是通过Python API还是命令行界面，都可以使用此功能。
若要启用多GPU训练，请指定您希望使用的GPU设备ID。

要使用2个GPU进行训练，请使用CUDA设备0和1，使用以下命令。根据需要扩展到更多GPU。

from ultralytics import YOLO


model = YOLO('model/yolov8n.pt')  


results = model.train(data='coco128.yaml', epochs=100, imgsz=640, device=[0, 1])

yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640 device=0,1

在处理深度学习模型时，从之前保存的状态恢复训练是一个关键特性。在各种情况下，这可能很方便，比如当训练过程意外中断，或者当您希望用新数据或更多时期继续训练模型时。

恢复训练时，Ultralytics YOLO将加载最后保存的模型的权重，并恢复优化器状态、学习率调度器和时期编号。这允许您无缝地从离开的地方继续训练过程。

在Ultralytics YOLO中，您可以通过在调用train方法时将resume参数设置为True并指定包含部分训练模型权重的.pt
文件路径来轻松恢复训练。

下面是使用Python和命令行恢复中断训练的示例：

from ultralytics import YOLO


model = YOLO('path/to/last.pt')  


results = model.train(resume=True)

yolo train resume model=path/to/last.pt

通过设置resume=True，train函数将从’path/to/last.pt’文件中存储的状态继续训练。如果省略resume
参数或将其设置为False，train函数将启动新的训练会话。

请记住，默认情况下，检查点会在每个时期结束时保存，或者使用save_period参数以固定间隔保存，因此您必须至少完成1个时期才能恢复训练运行。

YOLO模型的训练设置是指用于对数据集进行模型训练的各种超参数和配置。这些设置会影响模型的性能、速度和准确性。一些常见的YOLO训练设置包括批大小、学习率、动量和权重衰减。其他可能影响训练过程的因素包括优化器的选择、损失函数的选择以及训练数据集的大小和组成。仔细调整和实验这些设置以实现给定任务的最佳性能是非常重要的。

在训练YOLOv8模型时，跟踪模型随时间的性能变化可能非常有价值。这就是记录发挥作用的地方。Ultralytics的YOLO提供对三种类型记录器的支持 -
Comet、ClearML和TensorBoard。

要使用记录器，请在上面的代码片段中的下拉菜单中选择它并运行。所选的记录器将被安装和初始化。

TensorBoard
是TensorFlow的可视化工具包。它允许您可视化TensorFlow图表，绘制有关图表执行的定量指标，并展示通过它的附加数据，如图像。

load_ext tensorboard
tensorboard --logdir ultralytics/runs  

在本地使用TensorBoard，运行下面的命令并在 http://localhost:6006/ 查看结果。

tensorboard --logdir ultralytics/runs  

这将加载TensorBoard并将其定向到保存训练日志的目录。

在设置好日志记录器后，您可以继续进行模型训练。所有训练指标将自动记录在您选择的平台中，您可以访问这些日志以监控模型随时间的表现，比较不同模型，并识别改进的领域。

将 YOLOv8n 模型导出为 ONNX 或 TensorRT 等不同格式。查看下面的参数部分，了解完整的导出参数列表。

from ultralytics import YOLO


model = YOLO('model/yolov8n.pt')  
model = YOLO('path/to/best.pt')  


model.export(format='onnx')

yolo export model=yolov8n.pt format=onnx  
yolo export model=path/to/best.pt format=onnx  

YOLO 模型的导出设置是指用于在其他环境或平台中使用模型时保存或导出模型的各种配置和选项。这些设置会影响模型的性能、大小和与不同系统的兼容性。一些常见的
YOLO 导出设置包括导出的模型文件格式（例如 ONNX、TensorFlow SavedModel）、模型将在哪个设备上运行（例如
CPU、GPU）以及是否包含附加功能，如遮罩或每个框多个标签。其他可能影响导出过程的因素包括模型用途的具体细节以及目标环境或平台的要求或限制。重要的是要仔细考虑和配置这些设置，以确保导出的模型针对预期用例经过优化，并且可以在目标环境中有效使用。

下表中提供了可用的 YOLOv8 导出格式。您可以使用 format 参数导出任何格式的模型，比如 format='onnx' 或 format='engine'。

Yolov8 训练代码+部署事例+参考手册 1G资料

下载链接：点击下载