1. 程式人生 > >windows10下安裝TensorFlow Object Detection API

windows10下安裝TensorFlow Object Detection API

安裝步驟:

1、下載原始碼後解壓,修改資料夾名為models

(以下步驟中涉及到路徑的地方需要根據自己的實際情況而定)

2、安裝protoc:

下載後複製到與models同名的資料夾下,解壓,生成:bin、include。將bin資料夾下的protoc.exe複製到C:\Windows\System32資料夾下。cmd開啟命令列介面,輸入命令protoc,出現如下介面說明安裝成功:

3、編譯proto檔案

在models/research下執行Windows PowerShell(注意,這裡必須是PowerShell,執行cmd會報錯),輸入如下命令:

Get-ChildItem object_detection/protos/*.proto | Resolve-Path -Relative | %{ protoc $_ --python_out=. }

執行完成後,可以檢查object_detection/protos/資料夾,如果每個proto檔案都成了對應的以py為字尾的python原始碼,就說明編譯成功了。

4、新增環境變數

在Anaconda\Lib\site-packages新增一個路徑檔案,如tensorflow_model.pth,必須以.pth為字尾,寫上你要加入的模組檔案所在的目錄名稱,如下圖: 

5、執行models/research下的setup.py

python setup.py build

python setup.py install

6、安裝完成測試

在models/research下執行如下命令:

python object_detection/builders/model_builder_test.py

出現如下資訊,說明已安裝成功:

7、執行已經訓練好的模型

在research資料夾下執行命令:jupyter notebook,接著在jupyter中開啟object_detection資料夾,並單擊object_detection_tutorial.ipynb執行試例檔案。如下所示:

執行所有程式碼後的結果如下: