安裝Google Object Dectection API
官網:https://github.com/tensorflow/models/tree/master/research/object_detection
- 安裝python3.5
- 使用virtualenv創建基於python3.5的虛擬環境
- 安裝tensorflow
- 安裝各種依賴
此處完全按照官網操作即可
sudo apt-get install protobuf-compiler python-pil python-lxml python-tk
sudo pip install Cython
sudo pip install jupyter
sudo pip install matplotlib
sudo pip install Cython
sudo pip install pillow
sudo pip install lxml
sudo pip install jupyter
sudo pip install matplotlib
下載model
訪問 https://github.com/tensorflow/models 下載到本地(我在阿里的機器上用git沒操作成功,直接在瀏覽器裏下載)
解壓上傳到服務器coco API
下載,上傳到服務器
編譯
把以下文件夾拷貝到model/research下編譯protobuf
到research目錄下執行以下- 編輯PYTHONPATH
- 測試
以上主要參考https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md 即可
識別測試
運行例子
- 在服務器上,進入object dectection目錄啓動jupyter-book
- 在本機打開notebook
把上圖的紅框裏的地址放到地址欄裏(要把0.0.0.0換成你機器的IP)
- 找到object_dectection_tutorial.ipynb
- 打開
- 運行:Cell、Run All
- 等十秒左右,最下面出現對示例圖片的識別
識別自己的圖片
看剛剛運行的代碼,找到test_images,這就是獲取圖片的位置
在服務器上找到test_images文件夾,會找到兩張自帶圖片,把自己的圖片拷貝進去,最好用相同的名字替代
-識別了幾張自己的,效果如下
可見缺省模型對骷髏的識別還不行:)