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
 

猜你喜欢

转载自blog.csdn.net/yuejisuo1948/article/details/91044355