将模型转换为适用于Coral Dev Board的TF Lite

时间:2019-05-30 15:40:05

标签: tensorflow tensorboard tensorflow-lite google-coral

通过遵循本教程,我为自定义语音识别开发了一个模型:https://www.tensorflow.org/tutorials/sequences/audio_recognition 我已经使用自定义参数自定义了模型,并且冻结了图表。现在,我想将此模型部署在Coral Dev Board上。由于这些原因,我进行了8位量化感知训练。但是,使用tflite_convert工具将冻结的图形转换为Tensorflow Lite模型时遇到了麻烦。命令:

 tflite_convert --output_file=model.tflite --graph_def_file=frozen.pb --input_arrays=wav_data --output_arrays=labels_softmax --inference_type=QUANTIZED_UINT8

返回以下错误:

  

ValueError:为输入数组'wav_data'提供输入形状。

如何找到所请求参数的正确值?任何想法? 谢谢。

1 个答案:

答案 0 :(得分:0)

我认为您可能使用了错误的输入节点,“ wav_data”是一个DecodeWav op,它接收.wav文件的内容,但是您可能想要传递从麦克风捕获的原始样本数据,而是放在“ decoded_sample_data”中。在这种情况下,以下是我通常要使用的参数:

--input_shapes=16000,1:1 --input_arrays=decoded_sample_data,decoded_sample_data:1
相关问题