当前位置: 代码迷 >> 综合 >> pytorch-yolov5如何训练自己的数据集,手把手从头教学
  详细解决方案

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

热度:72   发布时间:2023-10-19 18:42:40.0

先下载源码

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这两个