先下载源码
这是下载好的代码,我们首先要做的是安装运行代码的环境,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这两个