- 前言
- 一、配置环境
- 二、下载源码和官方权重
- 三、开始训练
前言
《学习笔记》由于最近在弄模型部署到移动端的事情，所以需要训练最新版本（6.1）的权重，于是我开始使用最新版的yolov5去训练自己的模型，当然最先开始用的是给的coco数据集。跳了两个bug，完美解决
一、配置环境默认你已经会配置环境啦(这一部分其实还包括在windows7上配置环境，当时走了很多坑)，cuda=11.0 python=3.7 torch=1.7.1 torchvision=0.8.2
二、下载源码和官方权重网站贴上，下载zip，解压到桌面
https://github.com/ultralytics/yolov5
进入如下网站，下载官方权重，我下载的yolov5s（下载哪个权重就是使用哪个模型训练，众所周知yolov5共有五个版本）并放入解压后的yolov5-master文件夹下
https://github.com/ultralytics/yolov5/releases/tag/v6.1三、开始训练
点击运行train.py，我的显卡不太好，显存也比较小，所以batchsize=3，在train.py的484行
代码如下（示例）：
parser.add_argument('--batch-size', type=int, default=3, help='total batch size for all GPUs, -1 for autobatch')
其他基本不要动，运行过程中dataests会自动生成在和yolov5-master同级目录下
