安裝執行谷歌開源的TensorFlow Object Detection API視訊物體識別系統
阿新 • • 發佈:2019-01-03
Linux安裝
參照官方文件:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md安裝Python
Anaconda下載
安裝TensorFlow
筆者用給pip方式安裝
安裝CUP版的TensorFlow:
pip install tensorflow
安裝CUP版的TensorFlow:pip install tensorflow
筆者選擇TensorFlow的版本為1.5.0,因此:
pip install tensorflow==1.5.0
安裝Protoc
Protoc用於編譯相關程式執行檔案,進入Protoc 下載頁,根據具體的作業系統選擇對應的版本,筆者為Win7下載類似下圖中帶win32的壓縮包,解壓後將bin資料夾內的protoc.exe拷貝到c:\windows\system32目錄下
筆者選擇的是:
因為最近的3.5.1存在BUG會操作Object Detection API無法編譯。
安裝git
進入git官網下載Windows平臺的git
下載程式碼並編譯
在cmd中輸入如下程式碼:
git clone https://github.com/tensorflow/models.git
或在官網,下載ZIP包:
cmd進入到models/research/資料夾,編譯Object Detection API的程式碼:
# From models/research/
protoc object_detection/protos/*.proto --python_out=.
將庫新增到PYTHONPATH,models/research和models/research/slim目錄應附加到PYTHONPATH中。
D:\models-master\research;D:\models-master\research\slim
測試安裝,通過執行以下命令來測試您是否正確安裝了Tensorflow Object Detection API:
python object_detection/builders/model_builder_test.py
執行notebook demo
在models-master\research檔案下執行以下命令:notebook demo
瀏覽器自動開啟,顯示如下介面:
選擇object_detection進入
選擇object_detection_tutorial.ipynb進入
點選Cell內的Run All,等待幾分鐘左右(下載模型),即可顯示如下結果:
至此官網示例已搭建執行完畢