(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文件