faster-cnn在pytorch上實現(一)——下載源碼

源碼安裝

準備

源碼

首先，先下載faster-rcnn源碼

git clone https://github.com/jwyang/faster-rcnn.pytorch.git

然後進入目錄，再創建一個data文件

cd faster-rcnn.pytorch && mkdir data

注意：pytorch只能安裝0.4.0及以下版本，像現在的1.0版本就不支持，可以看我上一篇博客把版本降下去。

數據集

• PASCAL_VOC 07+12

• COCO

• Visual Genome
  可以自己在網上下載和解壓

編譯安裝

cd lib
sudo sh make.sh

模型下載（vgg16）

下载预训练模型
(这里所说的预训练模型不同于一般意义上迁移学习中的预训练模型，其实是已经在faster rcnn上训练好直接可以用的）
要的可以私我，我發鏈接給你

下一篇講訓練模型和測試