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