Question

我想使用Tensorflow将冻结的pb转换为tflite，但实际上网络的输入类型为“ [无，无，无，3]”，因此不支持。如何解决这个问题呢？这是错误：

/**
 *  Copyright © 2018  XYZ,
 *  All rights reserved.
 *  some XYZ PROPRIETARY/CONFIDENTIAL. 
 */
 package org.xxx.admin.controller.alerts;

/**
 * @author Harisha N
 */
public class Qwerty{

}

Answer 1

创建模型时，需要定义input_shape大小。例如，如果您有一批1和300 x 300 rgb的图像，则需要将输入形状的大小定义为1 x 300 x 300 x 3（批处理x宽度x高度x rgb）。

Answer 2

!tflite_convert \
  --output_file='/content/models/research/fine_tuned_model/frozen_inference_graph.tflite' \
  --graph_def_file='/content/models/research/fine_tuned_model/frozen_inference_graph.pb' \
  --inference_type=TENSORFLOW_GRAPHDEF \
  --input_arrays='image_tensor' \
  --output_arrays='num_detections', 'detection_boxes', 'detection_scores','detection_classes' \
  --mean_values=128 \
  --std_dev_values=128 \
  --input_shapes=1,300,300,3 \
  --change_concat_input_ranges=false \
  --allow_nudging_weights_to_use_fast_gemm_kernel=true \
  --allow_custom_ops

这可能会对您有所帮助。您只需要更改input_array和output_arrays的值

Tensorflow：如何转换为tflite？

2 个答案: