要使用COCO8数据集训练YOLO11模型,可以通过Python或命令行接口(CLI)进行。以下是如何开始训练的示例代码。对于可用参数的完整列表,请参考模型训练页面。
from ultralytics import YOLO
# 加载模型
model = YOLO(
"yolo11n.pt"
)
# 加载预训练模型(推荐用于训练)
# 训练模型
results = model.train(
data="coco8.yaml",
epochs=100,
imgsz=640
)
# 从预训练的*.pt模型开始训练
yolo detect train data=coco8.yaml model=yolo11n.pt epochs=100 imgsz=640
通过上述代码,可以加载一个预训练的YOLO模型,并使用COCO8数据集进行训练。训练过程中,可以通过调整epochs和imgsz参数来控制训练的轮数和图像大小。