将float类型的.pb模型量化成uint8类型的.tflite模型

最新推荐文章于 2024-12-08 18:25:56 发布

原创

最新推荐文章于 2024-12-08 18:25:56 发布 · 682 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow #深度学习 #人工智能

本文档介绍了如何将基于TensorFlow的float类型.pb模型转换为uint8类型的.tflite模型，详细阐述了转换过程中涉及的命令参数，包括输入文件路径、输出文件路径、输入格式、归一化参数、输入形状和数组等关键信息。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、转化命令


bazel run -c opt \
  //tensorflow/lite/toco:toco -- \
  --input_file=/home/yunji.he/code/tensorflow/tensorflow-1.15.5/pb_files/one_conver.pb  \
  --output_file=/home/yunji.he/code/tensorflow/tensorflow-1.15.5/pb_files/tflite_files/one_conver.tflite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --mean_value=128 \
  --std_value=98.80937957763672\
  --input_type=QUANTIZED_UINT8 \
  --inference_type=QUANTIZED_UINT8 \
  --input_shape=1,112,112,16 \
  --input_array=input \
  --output_array=outputs

 (格式有点乱，可以照着图修)

在这里插入图片描述