tensorflow pb模型量化

1：下载解压tensorflow源码，并进入tensorflow目录。

2：编译量化脚本（一定要加sudo）

sudo bazel build tensorflow/tools/quantization:quantize_graph

3：开始量化.pb文件

sudo bazel-bin/tensorflow/tools/quantization/quantize_graph--input=/tmp/PNet_output.pb --output_node_names="bbox_pred_out,cls_prob_out,landmark_pred_out" --output=/tmp/newpb/PNet_output.pb --mode=eightbit

如果过程中出现以下错误，逐一解决：

ImportError: No module named enum
pip install enum34

ImportError: No module named mock

pip install mock

更多的参数解释可以用命令：bazel-bin/tensorflow/tools/quantization/quantize_graph --help

参考书籍：Tensorflow技术解析与实战

文章来源: tensorflow pb模型量化