1. 程式人生 > >TensorFlow Object Detection API 技術手冊(2)——安裝TensorFlow Object Detection API

TensorFlow Object Detection API 技術手冊(2)——安裝TensorFlow Object Detection API

(一)下載TensorFlow模型原始碼

(二)將原始碼匯入Anaconda

開啟壓縮包找到research檔案,我們所需要的object detection的原始碼就在research中,如圖所示: 在這裡插入圖片描述 將research資料夾解壓到 (你的安裝位置/Anaconda3/Lib/site-packages)下,如圖所示: 在這裡插入圖片描述 在這裡插入圖片描述

(三)編譯proto檔案

GitHub下載protobuf包: 在這裡插入圖片描述 在這裡插入圖片描述 下載完成後,將壓縮包內(/bin)目錄下的protoc.exe檔案放到(research/)目錄下。 在這裡插入圖片描述 在(research/)目錄下按住shift鍵,點選右鍵開啟powershell視窗,輸入命令:

>>>.\protoc.exe .\object_detection\protos\*.proto --python_out=.(注意前後都有一個點)

在這裡插入圖片描述 編譯成功不會顯示其他資訊,進入(\research\object_detection\protos)目錄,發現所有.proto檔案已被編譯為.py檔案。 在這裡插入圖片描述

(四)裝載research模組

開啟Anconda Prompt,轉到(\research)目錄下,輸入命令:

>>>python setup.py install

閃出很多行程式碼後出現如下介面: 在這裡插入圖片描述

(五)新增Slim環境變數

在(**\Anaconda3\Lib\site-packages**)目錄下新建文字文件,輸入: 在這裡插入圖片描述

儲存並重命名為tensorflow_model.pth,如圖所示: 在這裡插入圖片描述 要注意檢視更改檔名後文件是否真的變為了.pth檔案,若重新命名沒有更改檔案型別,請對以下選項進行勾選: 在這裡插入圖片描述

(六)測試API

開啟Anconda Prompt,轉到(\research)目錄下,輸入命令:

>>>python object_detection/builders/model_builder_test.py

如圖所示: 在這裡插入圖片描述 出現上圖說明API已安裝成功。 下一節我們開始執行模型原始碼中自帶的示例程式。