tensorflow model安装

很久才发现tensorflow官网[https://github.com/tensorflow/models]提供了各种模型。
其中要求tf的版本在1.0以上。
以object_detection为例子。[https://github.com/tensorflow/models/tree/master/object_detection]

安装过程:

1.安装tensorflow
2.安装对应依赖库

  • Protobuf 2.6
  • Pillow 1.0
  • lxml
  • tf Slim (which is included in the "tensorflow/models" checkout)
  • Jupyter notebook
  • Matplotlib

安装 程序:

sudo apt-get install protobuf-compiler python-pil python-lxml
sudo pip install jupyter
sudo pip install matplotlib
#或者,protobuf之后的无法安装
sudo pip install pillow
sudo pip install lxml
sudo pip install jupyter
sudo pip install matplotlib

使用protobuf生成按照要求文件

#在 tensorflow/models/路径下运行
protoc object_detection/protos/*.proto --python_out=.

添加系统环境变量:

# pwd为tensorflow/models/
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

测试安装是否完成

python object_detection/builders/model_builder_test.py

如果成功则显示:

.......
----------------------------------------------------------------------
Ran 7 tests in 0.022s

OK

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容