由於本人是一個新鳥才疏學淺,在學習過程遇到很多問題,寫本博客的目的一是爲了記錄一下自己的學習歷程,另外一個是想給遇到同樣問題的童鞋提供一個思路。
一、安裝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檢測自己的圖片,如果有需要的朋友可以持續關注。
特意補圖的: