- 在官网克隆源码。
- 安装依赖,pip install -r requirements.txt
- 准备数据集,图片jpg 标签txt
- 将数据集划分为训练和验证集,数据放在和train.py同级目录
- 选模型,在model里选一个yolov5模型,复制粘贴并重命名为你数据集相关的名字
以上准备好模型和数据集了
- 找到data下的voc.yaml,复制粘贴并重命名为你数据集相关的名字,打开文件注释掉download: bash data/scripts/get_voc.sh;更改数据集地址;nc改成需要识别的数目;name填入类别名字
- 修改模型参数,打开第5步你的模型,改nc参数
- Train.py文件中改权重,改模型地址,改数据地址,改训练轮次,改batchsize,workers为电脑核数·
以上参数调整完了 运行train即可开始训练
- 输入tensorboard --logdir=runs/train可查看曲线