Tensorflow object_detection API 目標檢測環境搭建

由於本人是一個新鳥才疏學淺,在學習過程遇到很多問題,寫本博客的目的一是爲了記錄一下自己的學習歷程,另外一個是想給遇到同樣問題的童鞋提供一個思路。

一、安裝tensorflow

建議直接安裝Anaconda3,裏面包含了python3.6和各種tensorflow所需要的依賴庫,比如numpy、lxml等庫就不需要自己再手動安裝了,因爲本來就是一個新鳥,儘量打包可以完成的事情就不要爲難自己了。

Anaconda3的官網:https://www.anaconda.com/download/

tensorflow的安裝可以參照tensorflow中文社區文檔:tensorflow安裝教程

tensorflow下載地址:TensorFlow下載頁

二、安裝Object Detection API

tensorflow object_detection API 的安裝可以參考github上面的安裝教程.

object_detection的地址爲:tensorflow/models直接下載到本地進行解壓。

三、安裝protobuf

打開anaconda3的anaconda Prompt在裏面輸入conda install protobuf=3.4,切記一定要安裝3.4的版本不要安裝protoc3.5版本,不然會出現下面的錯誤:

找不到object_detection文件夾。具體原因見這裏:https://github.com/tensorflow/models/issues/2930

protobuf 3.4.0下載地址:https://github.com/google/protobuf/releases/tag/v3.4.0

將protoc-3.4.0-win32.tar.gz進行解壓,將bin裏面的protoc.exe拷到models-master\research中,然後按住shift鍵,右擊鼠標右鍵:點擊在此處打開命令窗口,在裏面輸入:protoc --python_out=. object_detection/protos/*.proto

停頓一到兩秒出現如下結果:

就說明protoc已經配置好。

四、測試API

1、在命令行內輸入:如下內容

protoc object_detection/protos/*.proto --python_out=.

如果這裏沒有報錯說明你上面配置完好;

2、安裝tensorflow model以及slim

首先在命令行內執行:python setup.py install 

只需等待幾秒就可以安裝成功。

這時候如果你執行:python object_detection/builders/model_builder_test.py將會報錯:

ImportError:No module named nets

這時候就說明你的slim還沒有安裝,進入到slim文件下:同樣執行

python setup.py install

就會出現如下情況:

說明你slim已經安裝成功,這時候需要把model和slim都配置到path立馬;在path原有配置情況下添加:

G:\models\models-master\research;G:\models\models-master\research\slim保存,這樣model和slim的環境變量就配置好了。

3、回到research目錄下執行如下命令:

Python object_detection/builders/model_builder_test.py

恭喜你測試成功。

4、通過jupyter_notebook查看訓練結果

這裏建議使用IE瀏覽器且是最新版本,我這裏使用的是IE11,如果瀏覽器版本太低的話,將會加

載不出來了;

執行命令:jupyter_notebook

最新版本已經換成:jupyter notebook

效果如下:

然後找到object_detection文件下的object_detection_tutorial.ipynb在瀏覽器中自動生成如下內容:

找到cell然後運行Run All,如果你的網速夠快電腦性能夠好,順利的情況下將會出現正常的結果,由於博主電腦接近報廢的情況,運行較慢,等以後有時間再把這張運行效果圖補回來。

後續內容將會是使用obect_detection API檢測自己的圖片,如果有需要的朋友可以持續關注。

特意補圖的:

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章