10min复现yolov5官方源码

---

前言

《学习笔记》由于最近在弄模型部署到移动端的事情，所以需要训练最新版本（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同级目录下