參考:https://www.jianshu.com/p/1ed2d9ce6a88
安裝
- 安裝conda+tensorflow庫
- 下載protoc linux x64版,https://github.com/protocolbuffers/protobuf/releases
- 將下載的zip文件,進入解壓后的文件,將里面的bin/protoc文件復制到/usr/bin/protoc
sudo cp bin/protoc /usr/bin/protoc
- 下載models倉庫
git clone https://github.com/tensorflow/models.git
- 在models/research/目錄下,運行
protoc object_detection/protos/*.proto --python_out=. python setup.py install
- 刪除models/research/slim/BUILD文件,並運行
python setup.py install
測試
- 運行測試腳本,結果出現OK字樣則表示正確。
python object_detection/builders/model_builder_test.py
- 在models下打開jupyter-notebook,並運行research/object_detection/object_detection_tutorial.ipynb,運行成功后出現下圖
jupyter notebook .