Tensorflow object detection API 搭建物體識別模型（一）

一、開發環境

　1）python3.5

　2）tensorflow1.12.0

　3）Tensorflow object detection API ：https://github.com/tensorflow/models下載到本地，解壓

　　我們需要的目標檢測代碼在models-research文件中：

 

　　其中object_detection中的README.md記錄了API的使用指導：

　　注意：models包含了眾多的模塊，可以根據需要選擇下載，也可以將整個models一塊兒下載，大概420M

　4）Protobuf 安裝與配置

　　Protobuf(Google Protocol Buffers)是GG開發的的一套用於數據存儲，網絡通信時用於協議編解碼的工具庫。和XML，Json數據差不多，把數據按某種形式保存起來。相對與XML和Json的不同之處，它是一種二進制的數據格式，具有更高的傳輸，打包和解包效率。

　　在https://github.com/google/protobuf/releases 網站中選擇windows 版本，解壓后將bin文件夾中的【protoc.exe】放到models中的research目錄下，執行下面的代碼，將object_detection/protos下的.proto文件轉換成.py文件

'''
需要到protoc所在的目錄下執行protoc
protoc object_detection/protos/*.proto --python_out=./
'''

import os
os.chdir( 'models-master/research' )

for each in os.listdir( 'object_detection/protos' ):
    if each.endswith('proto'):
        os.system('protoc object_detection/protos/%s --python_out=.' % each)

　5）jupyter notebook下運行實例

　　在object_detection那輸入cmd命令，然后運行jupyter notebook

 

　　打開如下文件，然后依次點擊Cell->Run All

　6）demo結果

　　如果代碼成功運行的話，運行結果如下圖所示：

　

　下一步我們可以在此基礎上對代碼進行適當的修改，可以用已有的模型來檢測自己的圖片，甚至視頻，並輸出結果。

　接着請讀者參考：Tensorflow object detection API 搭建物體識別模型（二）