windows10下安裝TensorFlow Object Detection API的步驟

安裝步驟：

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

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

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

2、安裝protoc：

打開網(wǎng)址https://github.com/protocolbuffers/protobuf/releases，，我這里下載的是proto-3.7.1-win64.zip，根據(jù)自己的電腦配置來。

下載后復(fù)制到與models同名的文件夾下，解壓，生成：bin、include。將bin文件夾下的protoc.exe復(fù)制到C:\Windows\System32文件夾下。cmd打開命令行界面，輸入命令protoc，出現(xiàn)如下界面說明安裝成功：

3、編譯proto文件

在models/research下運行Windows PowerShell（注意，這里必須是PowerShell，運行cmd會報錯），輸入如下命令：

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

運行完成后，可以檢查object_detection/protos/文件夾，如果每個proto文件都成了對應(yīng)的以py為后綴的python源碼，就說明編譯成功了。

4、添加環(huán)境變量

在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

出現(xiàn)如下信息，說明已安裝成功：

7、執(zhí)行已經(jīng)訓(xùn)練好的模型

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

運行所有代碼后的結(jié)果如下：

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。