android - 将 .pb 转换为 .tflite 时反量化的自定义实现

Question

我正在尝试使用 toco 将 tensorflow lite 量化的 .pb 文件转换为 .lite。创建 .pb 文件的命令是:
retrain.py 是 here和 here .

python retrain.py \
--bottleneck_dir=/mobilenet_q/bottlenecks \
--how_many_training_steps=4000 \
--output_graph=/mobilenet_q/retrained_graph_mobilenet_q_1_224.pb \
--output_labels=/mobilenet_q/retrained_labels_mobilenet_q_1_224.txt \
--image_dir=/data \
--architecture=mobilenet_1.0_224_quantized

当我尝试使用 toco 命令将 .pb 文件转换为 .tflite 时:

bazel run --config=opt //tensorflow/contrib/lite/toco:toco \
  -- --input_file= retrained_graph_mobilenet_q_1_224.pb \
  --output_file= retrained_graph_mobilenet_q_1_224.lite \
  --input_format=TENSORFLOW_GRAPHDEF \
  --output_format=TFLITE \
  --input_shape=1,224,224,3 \
  --input_array=input \
  --output_array=final_result \
  --inference_type=FLOAT \
  --input_data_type=FLOAT

我收到错误:标准 TensorFlow Lite 运行时不支持模型中的某些运算符。如果您有针对它们的自定义实现，您可以使用 --allow_custom_ops 禁用此错误，或者在调用 tf.contrib.lite.toco_convert() 时设置 allow_custom_ops=True。以下是您需要自定义实现的运算符列表:Dequantize。

我在 github 和 stackoverflow 上搜索过，但没有找到满意的答案。

Answer 1

讨论和解决方案是here .