①首先要从GitHub/tensorflow下载models包,
git clone https://github.com/tensorflow/models
如果没安装过git的话,可以用conda安装
conda install git
②安装protoc
从 https://github.com/google/protobuf/releases 下载 protoc-3.5.0-win32.zip 并解压,将bin文件夹下的执行文件protoc拿到models/research目录下。
在cmd命令行下 :
L:
cd tensorflow\models\research
protoc object_detection\protos\*.proto --python_out=.
然后要记得安装几个包matplotlib, PIL(特别注意现在不要用 pip install PIL安装,要用pip install Pillow),cv2(pip install opencv-python)
③测试
L:\TensorFlow\models\research\object_detection\builders
python model_builder_test.py
若结果如下,则安装成功。
Ran 11 tests in 0.061s
OK
遇到的一个坑是ImportError: No module named 'object_detection',原因是slim包的目录没有加到python的路径中。我的解决方法是直接在测试程序中加了几句代码:
import sys
sys.path.append('L:\\TensorFlow\\models\\research')
sys.path.append('L:\\TensorFlow\\models\\research\\slim')
未完成部分。
测试
在anaconda的TensorFlow环境下打开cmd,输入
jupyter notebook L:\tensorflow
然后找到object_detection\object_detection_tutorial.ipynb, 执行run all