1. 程式人生 > >tensorflow .pb檔案模型量化

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檔案