- 下載項目
```
git clone https://github.com/endernewton/tf-faster-rcnn.git
```
- 編譯,配置相應(yīng)的運行環(huán)境
- 安裝Cpython (這里相應(yīng)的模塊需要根據(jù)個人平時的使用下載,有很多的本人電腦已經(jīng)存在,便沒有一一列舉出來)
? ? ? ? ```
? ? ? ? ? sudo apt-get install cpython
? ? ? ? ? ```
- 修改項目,改為cpu運行
【1】打開lib->fast_rcnn->config.py,查看第236行,將True修改為False.
作用:禁用gpu,使用cpu進(jìn)行計算
【2】打開lib->fast_rcnn->nms_wrapper.py,進(jìn)行注釋,第12行
【3】打開lib->setup.py,進(jìn)行注釋,第55行,第120-136行
- 進(jìn)入項目的lib目錄中執(zhí)行```make ```
- 安裝Python COCO API,使用coco數(shù)據(jù)集
```shell
cd data
git clone https://github.com/pdollar/coco.git
cd coco/PythonAPI
make
```
- 運行demo和訓(xùn)練
- 下載預(yù)訓(xùn)練模型[voc_0712_80k-110k.tgz](https://pan.baidu.com/s/1kWkF3fT )
放在data文件目錄下,解壓.
- 建立訓(xùn)練模型的軟連接,data文件目錄下進(jìn)行
```shell
NET=res101
TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
mkdir -p output/${NET}/${TRAIN_IMDB}
cd output/${NET}/${TRAIN_IMDB}
ln -s ../../../voc_2007_trainval+voc_2012_trainval ./default
cd ../../..
```
- 運行demo
data目錄文件下
```python
../tools/demo.py
```