代码来源于Github:https://github.com/eriklindernoren/PyTorch-YOLOv3 环境配置:Windows 10+CUDA 9.2+PyTorch 1.1.0+python 3.6.2

网上关于环境配置有很多教程,这里不再赘述。环境配置成功,接下来就是按照代码包中READEME.md进行安装。README.md安装部分内容如下:

pytorchyolov3 pytorchyolov3环境搭建_bash


一、Clone and install requirements(下载代码和一些需要的包)

1.git clone https://github.com/eriklindernoren/PyTorch-YOLOv3(下载代码)

首先利用命令行进入到你想要下载代码的文件位置,接着利用指令下载代码。以本人为例,按照如下操作即可(前提是电脑已经安装了git)。

pytorchyolov3 pytorchyolov3环境搭建_github_02


2.cd PyTorch-YOLOv3/(cd到该目录)

sudo pip3 install -r requirements.txt(安装requirement文件里指定的各种包)这一步应该在Ubuntu系统一个指令就搞定了,但是现在只能自己一个一个下载了。

pytorchyolov3 pytorchyolov3环境搭建_github_03


打开requirement.txt,有如下9个包:

pytorchyolov3 pytorchyolov3环境搭建_pytorchyolov3_04


我用的anaconda,就是进入到PyTorch环境中,利用如下指令挨个安装

conda install numpy

二、Download pretrained weights(下载预训练模型)

1.$ cd weights/

首先cd到weights文件夹下,

2.$ bash download_weights.sh

windows不能执行.sh文件,因此要借助git bash来进行。可以这么做(这里要注意git bash要cd只能一步一步来),以我举例。

pytorchyolov3 pytorchyolov3环境搭建_bash_05


你是不是以为接下来只要输入bash download_weights.sh就大功告成啦!!!

您又错了,貌似是git只能下载github的东西,我也不是很清楚。我就按照download_weights.sh文件的网址挨个迅雷下载了,并且放到该文件夹下。

这里科普一下,关于git bash的复制、粘贴不再是Ctrl+C和Ctrl+V了,建议您使用右键去复制粘贴,加快速度。

三、Download COCO(下载COCO数据集,这一步搞了好久)

1.$ cd data/

2.$ bash get_coco_dataset.sh(这一步也不能git bash来实现,中间有几步下载的东西不在Github)

打开get_coco_dataset.sh文件

pytorchyolov3 pytorchyolov3环境搭建_git_06


第一步、Clone COCO API

cd到data文件夹下,输入指令git clone https://github.com/pdollar/coco

完成后,data文件夹会多出一个coco文件夹;再cd到coco文件夹下;mkdir images以为创建一个名为images的文件夹;cd images即在cd到images文件夹下。

第二步、Download Images

从指定网址下载数据集并且解压到images文件夹下(解压的时间可能会久一点);

cd…即返回上一目录,到coco文件夹下;

第三步、Download COCO Metadata

从四个网址下载4个文件至coco文件夹下,其中两个压缩包,两个part文件,将压缩包进行解压;

第四步、Set Up Image Lists(个人觉得就是建立数据集的存放路径,paste是个合并指令)

5k.part中内容如下

pytorchyolov3 pytorchyolov3环境搭建_bash_07


还是使用git bash cd至coco文件夹下,依次输入

paste <(awk "{print \"$PWD\"}" <5k.part) 5k.part | tr -d '\t' > 5k.txt
paste <(awk "{print \"$PWD\"}" <trainvalno5k.part) trainvalno5k.part | tr -d '\t' > trainvalno5k.txt

之后就会生成5k.txt和trainvalno5k.txt两个文件夹,如下图:

pytorchyolov3 pytorchyolov3环境搭建_github_08


但是现在test.py还是使用不了,原因就在于windows和linux路径的不同。因此这里需要将上图.txt文件中的/d替换为D:即可大功告成啦!!!

至此,## Installation部分就挖成啦,接下来就可以运行test.py啦!!!

如有问题,可与我联系。