UEditor 进程 Nodepad 阿里巴巴 微信商家收款 networking Select2 vue配置 bootstrap框架 jquery循环 两个正态分布相乘 鼠标失去焦点事件 matlab复数求模 python正则表达式 python字典get python命令行参数 配置python环境 python创建文件 python如何定义变量 python传参 java入门教程 java文件写入 java中的队列 java初级入门教程 java正则匹配数字 java可变参数 java接口规范 php网络编程 hadoop权威指南 id解锁大师 视频字幕提取器 银头鲑鱼 系统工具箱 HTML5从入门到精通 bz2解压命令 掌门一对一下载 苹果手机验机软件 小米8游戏模式 关闭页面 pro换肤
当前位置: 首页 > 学习教程  > 编程语言

tensorflow pb模型的量化

2020/8/31 14:03:30 文章标签:

环境:ubuntu 16.04
tensorflow1.12.0

tensorflow源码1.10和1.12量化的脚本不一样,我以前用的1.10,我还存着使用方法,但是我现在就想量化1.12,好折腾一女的,然后今天又折腾了一波这玩意!
先说tensorflow1.10:

官网:
https://github.com/tensorflow/tensorflow/tags

进入文件夹,先编译bazel-bin

bazel build tensorflow/tools/quantization:quantize_graph

量化pb

bazel-bin/tensorflow/tools/quantization/quantize_graph --input=*.pb --output_node_names="*" --output=*.pb --mode=eightbit --logtostderr
--input : 要量化的模型pb文件(完整路径)
--output_node_names : 输出的tensor节点名字
--output : 输出的.pb文件(完整路径)

至此,一个完整的流程就完成了!

tensorflow1.12 tensorflow/tools/quantization/下没有quantize_graph,已经废弃了,需要编译transform_graph工具才可以

bazel build tensorflow/tools/graph_transforms:transform_graph

在这里插入图片描述
出现这样的界面就是编译成功了!
量化pb模型

bazel-bin/tensorflow/tools/graph_transforms/transform_graph \
--in_graph=*.pb \
--out_graph=*.pb \
--outputs="*" \
--transforms='quantize_weights'

注意:
--in_graph : 要量化的模型pb文件(完整路径)
--out_graph : 输出的.pb文件(完整路径)
--outputs : 输出的tensor节点名字

我看网上有人说可以用summarize_graph分析量化前后的模型区别,权重量化、模型减小、增加了一些和量化和反量化的节点,所以我又编译了一下summarize_graph

bazel build tensorflow/tools/graph_transforms:summarize_graph

在这里插入图片描述
出现这样的界面就是编译成功了!

bazel-bin/tensorflow/tools/graph_transforms/summarize_graph \
--in_graph=*.pb \
--print_structure=true
注意:
--in_graph : 要量化的模型pb文件(完整路径)

本文链接: http://www.dtmao.cc/news_show_150155.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?