tensorflow 模型量化

文章参考https://blog.csdn.net/u012101561/article/details/82587806?tdsourcetag=s_pcqq_aiomsg

 

tensorflow的量化有很多种方式，有些需要用到tflite，有些不需要，下面介绍一下不需要tflite也能量化的方式：

在较早的版本上有官方的一个量化脚本，比如tensorflow-1.10.0\tensorflow\tools\quantization\quantize_graph.py

 

假设你想把32bit浮点型的a.pb量化成8bit整型，命名为b.pb，只需在命令行中输入：

python  quantize_graph.py  --input=a.pb  --output_node_names=out1,out2,out3  --output=b.pb  --mode=eightbit