tf-faster-rcnn[cpu]實現(xiàn)目標(biāo)檢測(一)項目環(huán)境配置

- 下載項目

```

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

```

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容