toco生成tflite模型

模型量化有两种方式:python API和toco工具,这两种方式都可以进行tflite模型转换,包括float16、int8两种数据格式的tflite模型

下面介绍使用toco工具进行模型转换的步骤:

1.从tensorflow官方clone tensorflow源码,我使用的是tensorflow==1.13

   git clone https://github.com/tensorflow/tensorflow.git

2.安装bazel工具,详细步骤可以根据bazel官网进行安装

   https://docs.bazel.build/versions/master/install-ubuntu.html

3.编译toco工具,编译方式

  bazel build //tensorflow/lite/toco:toco

4.量化训练后的模型(带有虚节点)生成float16tflite模型

./bazel-bin/tensorflow/lite/toco/toco --input_file=model.pb   --output_file=pl.tflite   --input_format=TENSORFLOW_GRAPHDEF   --output_format=TFLITE   --inference_type=FLOAT --input_shape=1,768,768,3   --input_array=Placeholder   --output_arrays=strided_slice_11,strided_slice_12   --drop_fake_quant   --mean_value=0   --std_value=1

5.量化训练后的模型(带有虚节点)生成int8 tflite模型

./bazel-bin/tensorflow/lite/toco/toco --input_file=pixel_link_quan_sigmoid_split64_big_768_x4.pb   --output_file=pl2.tflite   --input_format=TENSORFLOW_GRAPHDEF   --output_format=TFLITE   --inference_type=QUANTIZED_UINT8   --input_shape=1,768,768,3   --input_array=Placeholder   --output_arrays=strided_slice_11,strided_slice_12   --drop_fake_quant   --mean_value=0   --std_value=1

参考:https://blog.csdn.net/gubenpeiyuan/article/details/79652227

           https://zhuanlan.zhihu.com/p/59496851

           https://stackoverflow.com/questions/48112174/tensorflow-lite-error-converting-to-tflite-using-toco

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值