pytorch-yolov5如何训练自己的数据集，手把手从头教学

先下载源码

这是下载好的代码，我们首先要做的是安装运行代码的环境，pytorch版本要选择pytorch1.7及以上版本，因为代码中有SiLU函数只有比较高的版本才能用，

pytorch下载及环境安装

cpu版本和gpu版本都是可以训练的，我直接有写过安装pytorch-gpu的教程，

大家可以看我之前的博客，在win10上安装pytorch-gpu版本

安装完pytorch版本之后，需要安装其他配套的库，直接

pip install -r requirements.txt

就可以自动安装相应的库，如果速度太慢了，可以在后面加上镜像源，比如

-i https://pypi.tuna.tsinghua.edu.cn/simple 

速度还是非常快的。

完成以上步骤之后，就需要自己来创建数据集了

数据集制作

在项目文件夹下创建VOCdevkit/VOC2007这两个