【Tensorflow】bazel编译tensorflow工具summarize_graph、freeze_graph、toco

环境：

Ubuntu 16.04

Anaconda3

Python3.6

Bazel 0.15

Tensorflow1.12

一、新建虚拟环境
先用anaconda新建一个专门编译这些工具的虚拟环境。Anaconda的安装请参考链接：【Anaconda】Ubuntu16.04 安装anaconda3。

conda create -n bazelBuild python=3.6
激活环境。

source activate bazelBuild
二、安装bazel
请参考链接：【Linux】Google bazel安装教程ubuntu16.04。

三、安装依赖项
    sudo apt install python3-dev python3-pip #或者python-dev python-pip
    pip install six numpy wheel setuptools mock future
    pip install  keras_applications==1.0.6 --no-deps
    pip install  keras_preprocessing==1.0.5 --no-deps
四、下载tensorflow源码
git clone  [email protected]:tensorflow/tensorflow.git
git checkout r1.12
cd tensorflow/
五、summarize_graph
summarize_graph可以查看网络节点，在只有一个固化的权重文件而不知道具体的网络结构时非常有用。

编译命令：

bazel build tensorflow/tools/graph_transforms:summarize_graph
六、freeze_graph
freeze_graph是模型固化工具。

编译命令：

bazel build tensorflow/python/tools:freeze_graph
七、toco
toco是用来生成一个可供TensorFlow Lite框架使用tflite文件。

编译命令：

bazel build tensorflow/contrib/lite/toco:toco
 
————————————————
版权声明：本文为CSDN博主「heiheiya」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/heiheiya/article/details/100072487