object_detection 能够对ssd_mobilenets进行训练,为了体验效果,对object_detection 进行了安装
1.安装依赖项
我的是ubuntu 14.04,故在终端中输入如下命令
sudo pip install pillow
sudo pip install lxml
sudo pip install jupyter
sudo pip install matplotlib
如果是ubuntu 16.04,可输入:
sudo apt-get install protobuf-compiler python-pil python-lxml
sudo pip install jupyter
sudo pip install matplotlib
2.编译protobuf
终端进入models根目录,输入
protoc object_detection/protos/*.proto --python_out=.
(默认已经安装protobuf)
3.添加库路径
终端进入models根目录,输入
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
每次终端都需要输入一次
4.验证
终端进入models根目录,输入
python object_detection/builders/model_builder_test.py
若显示OK,则已经成功安装