通过遵循本教程,我为自定义语音识别开发了一个模型: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'提供输入形状。
如何找到所请求参数的正确值?任何想法? 谢谢。
答案 0 :(得分:0)
我认为您可能使用了错误的输入节点,“ wav_data”是一个DecodeWav op,它接收.wav文件的内容,但是您可能想要传递从麦克风捕获的原始样本数据,而是放在“ decoded_sample_data”中。在这种情况下,以下是我通常要使用的参数:
--input_shapes=16000,1:1 --input_arrays=decoded_sample_data,decoded_sample_data:1