1. 安装环境

    pip install ultralytics  # YOLOv8/v11
    # 或
    git clone https://github.com/ultralytics/yolov5
    pip install -r requirements.txt
    
  2. 准备数据集

    • 格式:每张图片对应一个 .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)
deviceGPU编号0(第一块GPU)
patience早停轮数50~100