tensorflow .pb文件模型量化

(1).下载解压tensorflow源码

(2).下载安装bazel工具,要求jdk不低于1.8

(3).编译量化脚本

sudo bazel build tensorflow/contrib/quantization/tools:quantize_graph

运行后报错:

ERROR: no such package 'tensorflow/contrib/quantization/tools': BUILD file not found on package path.

修改命令为:

sudo bazel build tensorflow/tools/quantization:quantize_graph

(4).生成.pb文件

(5).量化.pb文件

sudo bazel-bin/tensorflow/tools/quantization/quantize_graph --input=**.pb --output_node_names="**" --output=**.pb --mode=eightbit

--input : 要量化的模型pb文件

--output_node_names : 输出的tensor节点名字

--output : 输出的.pb文件