Yolov5 机器识别 模型训练

前期准备

1. yolov5 安装
2. 数据集准备

标柱训练集

使用Lableimg进行训练

1. pip 安装labelImg

    pip install labelImg
   

2. 打开labelImg

    (yolov5) ➜  ~ labelImg
   

3. 基础设置

• 选择模型输出
  
• 实用性配置
  

4. 开始标注
   
   快捷键：
   
   创建区块后设置标签
   
   标注完成后 Alt + s保存后即可在标签文件夹看到结果
   

开始训练

python train.py --img-size 640 --batch 24 --epochs 300 --data eggs.yaml --weights yolov5s.pt

• epochs: 训练次数
• data: 模型配置
• weights: 预训练模型