yolov4的Pytorch(cpu版)识别物体

目录

  • 一、准备文件
    • 1、pytorch-YOLOv4代码下载
    • 2、权重(weights)模型下载
    • 2、下载pytorch
  • 二、测试

一、准备文件

1、pytorch-YOLOv4代码下载

https://github.com/Tianxiaomo/pytorch-YOLOv4

2、权重(weights)模型下载

通过如下百度网盘链接进行权重文件下载,下载链接如下:(提取码:g0ju
https://pan.baidu.com/s/1ZNK7FbqC2TlVKukQFqGYSw
或者谷歌云盘下载
https://drive.google.com/open?id=1cewMfusmPjYWbrnuJRuKhPMwRe_b9PaT
把上面的代码下载之后解压,解压之后的目录下创建一个weight的文件夹,在weight的文件夹里放上面的权重文件。
yolov4的Pytorch(cpu版)识别物体_第1张图片

2、下载pytorch

进入下面这个网址
https://pytorch.org/
选择
yolov4的Pytorch(cpu版)识别物体_第2张图片
然后拉到下面,根据自己的需求选择对应的选项
yolov4的Pytorch(cpu版)识别物体_第3张图片
选好之后复制下面的命令在虚拟机上下载即可
yolov4的Pytorch(cpu版)识别物体_第4张图片

二、测试

进入到你解压的文件目录下,然后用如下命令,确保每一个文件都在对应的文件夹下。

python demo.py -cfgfile cfg/yolov4.cfg -weightfile weight/yolov4.weights -imgfile  data/dog.jpg

在这里插入图片描述
文件夹下面就多了一个图片文件
yolov4的Pytorch(cpu版)识别物体_第5张图片
再换一下选择的图片,记得把之前得到的第一张图重命名或者修改代码得到的文件名,否则会报错。
yolov4的Pytorch(cpu版)识别物体_第6张图片
得到的两张图如下
yolov4的Pytorch(cpu版)识别物体_第7张图片yolov4的Pytorch(cpu版)识别物体_第8张图片
参考链接
https://blog.csdn.net/qq_36693723/article/details/105907174

你可能感兴趣的:(pytorch)