您的位置:首页 > 其它

tensorflow .pb文件模型量化

2018-01-18 14:42 676 查看
(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文件
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: