1. 程式人生 > >安裝執行谷歌開源的TensorFlow Object Detection API視訊物體識別系統

安裝執行谷歌開源的TensorFlow Object Detection API視訊物體識別系統

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,等待幾分鐘左右(下載模型),即可顯示如下結果:


至此官網示例已搭建執行完畢