DataType error: cannot resolve DataType of [[[D

问题描述:
在安卓里调用tflite模型时,运行到tfLite.run(input, output);这句就会报错。提示如题所示错误。

原因分析:
在安卓调用tflite模型时,模型输入与模型输出都要与在python里训练时的输入输出相对应。我的模型在python里的输入形式是[1, 32, 256, 1],类型是float32,则在JAVA里也要是这种类型的ByteBuffer。
输出是[32,1,5116],类型是float32,则在java里接收模型输出的变量也应该是float类型的,可为float output[][][] = new float[32][1][5116];写成double类型的就会报如题的错误,数组中后面两维的数字[1][5116]都要写全。

猜你喜欢

转载自blog.csdn.net/u011208984/article/details/109741274