-
安装环境
pip install ultralytics # YOLOv8/v11 # 或 git clone https://github.com/ultralytics/yolov5 pip install -r requirements.txt -
准备数据集
-
格式:每张图片对应一个
.txt标注文件(YOLO格式:class x_center y_center width height,归一化值)。 -
目录结构示例:
dataset/ images/ train/ # 训练图8/10 val/ # 验证图2/10 labels/ train/ # 训练图对应标注 val/ # 验证图对应标注 -
创建
data.yaml:yaml
path: ../dataset train: images/train val: images/val nc: 2 # 类别数 names: ['cat', 'dog']
-
开始训练
yolo task=detect mode=train data=data.yaml model=yolov8n.pt epochs=100 imgsz=640
| 参数 | 含义 | 常见值 |
|---|---|---|
epochs | 训练轮数 | 100~300 |
imgsz | 输入图片尺寸 | 640 |
batch | 批量大小 | 根据显存调整(16, 32, 64) |
lr0 | 初始学习率 | 0.01(YOLOv5) |
device | GPU编号 | 0(第一块GPU) |
patience | 早停轮数 | 50~100 |