windows下Pytorch搭建YoloV4目标检测平台（参照Bubbliiiing博主文章运行成功）

本文是个人总结性文章
我看的是Bubbliiiing这个博主（大佬）的博客

这是原文地址：
https://blog.csdn.net/weixin_44791964/article/details/106214657?spm=1001.2014.3001.5501

所需环境：

这是提前准备好的环境
cuda==10.0
cudnn-10.0-windows10-x64-v7.4.1.5
python=3.6

---

scipy==1.2.1
numpy==1.17.0
matplotlib==3.1.2
opencv_python==4.1.2.30
tqdm==4.60.0
Pillow==8.2.0
h5py==2.10.0
在虚拟环境中直接安装就行，不会报错或者出问题。
如果出问题大概率是网络波动，多尝试几次
例如：pip install scipy==1.2.1

---

torch==1.2.0 
torchvision==0.4.0
需要单独下载对应的版本手动安装。
手动安装命令为：pip install 文件全名

torch下载链接：
https://download.pytorch.org/whl/torch_stable.html

---

配置环境后记得下载权重文件(weight.pth),分为coco和voc数据集，下载相应的版本。

相应的代码，权重文件，训练集下载地址：
https://github.com/bubbliiiing/yolov4-pytorch

下载好相关代码后需要建立虚拟环境：

创建虚拟环境（仅需创建一次）
conda create -n yolo4 python=3.6
激活虚拟环境
activate yolo4

所有环境的搭建需要在虚拟环境中完成。
上面提到的库都要安装到该虚拟环境下，才能保证程序的正常运行。

准备工作完成后可以进行预训练。
预训练之前需要仔细查看README.md文件（该文件在下载的源代码文件中有，仔细找找能看到）。

大部分问题在于环境配置上，如果不是环境配置问题就是yolo.py里面对应的类的名称出现问题。
出现：

RuntimeError: Error(s) in loading state_dict for YoloBody

看下面这个链接：https://blog.csdn.net/impact_10/article/details/118752248?spm=1001.2014.3001.5502
能解决部分问题

严格按照上述的描述来进行搭建，搭建过程比较顺利，本人也是一个新手，大家评论区相互交流沟通。👍👍👍👍