windows10下安裝TensorFlow Object Detection API的步驟

目錄

1、下載源碼後解壓，修改文件夾名爲models

2、安裝protoc：

3、編譯proto文件

4、添加環境變量

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

6、安裝完成測試

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

---

這篇文章主要介紹了windows10下安裝TensorFlow Object Detection API的步驟，小編覺得挺不錯的，現在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

安裝步驟：

模型源碼：https://github.com/tensorflow/models

1、下載源碼後解壓，修改文件夾名爲models

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

2、安裝protoc：

打開網址https://github.com/protocolbuffers/protobuf/releases，，我這裏下載的是proto-3.7.1-win64.zip，根據自己的電腦配置來。

下載後複製到與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

在models/research下運行如下命令：

• python setup.py build
• python setup.py install

6、安裝完成測試

python object_detection/builders/model_builder_test.py

出現如下信息，說明已安裝成功：

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

在research文件夾下運行命令：jupyter notebook，接着在jupyter中打開object_detection文件夾，並單擊object_detection_tutorial.ipynb運行試例文件。如下所示：

運行所有代碼後的結果如下：

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。