Faster-RCNN-TensorFlow-Python35
最近正在學(xué)習(xí)Faster-RCNN,環(huán)境歷經(jīng)一天時(shí)間終于成功安裝,借此記錄下整體安裝過程
本教程是Windows 10 + Python35 + CUDA 10.0 + cudnn 7.4.1.5 + tensorflow-gpu 1.13.2環(huán)境的配置過程
所使用的軟件包括
名稱 |
版本 |
CUDA |
10.0 |
CUDNN |
7.4.1.5 |
Anaconda3 |
4.2.0 |
Pycharm |
2019.3.5 |
整體過程中所需要的軟件包我都放在了網(wǎng)盤之中,需要可自?。ㄈ羰?,請(qǐng)留言)
軟件安裝包 提取碼:iskr
軟件安裝
VS2019
Tip:若已安裝VS2015且已安裝C++組件可跳過VS的安裝
Tip:若已安裝VS2017建議卸載后安裝
也可不卸載,先跳過這一步,之后如果出現(xiàn)問題可返回重做
打開安裝包,最好不要更改安裝路徑,在選擇組件時(shí)如下圖所示

點(diǎn)擊安裝即可
CUDA 10.0
打開安裝程序,如下圖所示

選擇自定義,然后下一步。

不做任何更改,下一步,直至安裝完成。
cudnn 7.4
下載壓縮包后解壓,復(fù)制圖中紅框所有文件 注意路徑

打開 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 路徑,粘貼到本目錄下,下圖為復(fù)制完成之后的情形

Anaconda3-4.2.0
安裝過程如圖所示

選擇 All Users,下一步

安裝路徑可以更改,也可不更改。
若更改路徑請(qǐng)看如下介紹,不更改可跳過
/*由于選擇的Anaconda3版本較低,所以不同于新版本的安裝*/
/*
eg:我選擇的路徑是E:\Program Files\
首先選中此目錄
然后復(fù)制Anaconda3到\后面 最終為E:\Program Files\Anaconda3
*/


選擇路徑之后下一步

勾選兩個(gè)框,然后install,安裝完成。
pycharm
默認(rèn)下一步即可,也可參考pycharm詳細(xì)安裝的博客
環(huán)境搭建
第三方庫安裝
Tips:若顯卡內(nèi)存小于4G,最好使用tensorflow-cpu
首先創(chuàng)建并激活環(huán)境
conda create -n tensorflow python=3.5
activate tensorflow
以下參考這位博主的文章
pip加速
打開C:\Users\用戶名
新建pip文件夾,打開新建pip.ini,右鍵選擇記事本打開,輸入以下代碼
[global]
index-url = http://pypi.mirrors.ustc.edu.cn/simple
[install]
use-mirrors =true
mirrors =http://pypi.mirrors.ustc.edu.cn/simple/
trusted-host =pypi.mirrors.ustc.edu.cn
關(guān)閉并保存
在桌面新建requirements.txt文件,輸入以下內(nèi)容
scipy==1.2.1
numpy==1.17.0
Keras==2.1.5
matplotlib
opencv_python==4.1.2.30
tensorflow_gpu==1.13.2
tqdm==4.60.0
Pillow
h5py
在cmd激活tensorflow環(huán)境的情況下,輸入如下指令
pip install -r C:\Users\ 用戶名 \Desktop\requirements.txt
用戶名更改位自己的即可
Tips 如果安裝過程中出現(xiàn)404,則使用如下命令
pip install numpy==1.16.5
pip install --upgrade pip
pip install --upgrade setuptools
pip install --no-cache-dir --force-reinstall -Iv grpcio==1.36.1
pip install tensorflow_gpu==1.13.2
然后再依次pip install requirements.txt文件中的包
逐個(gè)安裝比較麻煩,也可使用pip install -r C:\Users\ 用戶名 \Desktop\requirements.txt嘗試安裝。
ps:我安裝到這里心態(tài)快崩了,所以忍著痛一個(gè)一個(gè)安裝,祈求成功,最終果然成功了啊φ( ̄0 ̄)
源碼運(yùn)行
下載VOC2007數(shù)據(jù)集和vgg16網(wǎng)絡(luò)模型
下載Faster-RCNN源碼,此源碼原址為https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3
數(shù)據(jù)集處理
如下圖所示,選中三個(gè)壓縮包,然后解壓到當(dāng)前文件夾,會(huì)形成藍(lán)色框標(biāo)記的VOCdevkit文件夾

然后將文件夾名更改為 VOCDevkit2007 并移動(dòng)至Faster-RCNN中data目錄下。

VGG16模型
解壓 vgg_16_2016_08_28.tar.tar
將 vgg_16.ckpt 更名為 vgg16.ckpt
在Faster-RCNN的data目錄下新建 imagenet_weights 文件夾,并將上面的 vgg16.ckpt 移動(dòng)至此文件夾中

編譯準(zhǔn)備
進(jìn)入Faster-RCNN/data/coco/PythonAPI目錄下,執(zhí)行
python setup.py build_ext --inplace
python setup.py build_ext install
然后進(jìn)入Faster-RCNN/lib/utils目錄下執(zhí)行文章來源:http://www.zghlxwxcb.cn/news/detail-441523.html
python setup.py build_ext --inplace
運(yùn)行
Run train.py文章來源地址http://www.zghlxwxcb.cn/news/detail-441523.html
到了這里,關(guān)于Faster-RCNN環(huán)境搭配及運(yùn)行教程的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!