错误: 原因:java.lang.IllegalArgumentException:无法在 FLOAT32 类型的 TensorFlowLite 张量和 java.lang.String 类型的 Java 对象(与 TensorFlowLite 类型 STRING 兼容)之间进行转换。
我已经从我的数据集构建了一个神经网络,并且有 2 层,然后我将模型保存为 h5,然后使用 tf.keras 模型和转换将其转换为 tflite,但是当我将其部署在应用程序中时,它给了我上面的内容错误
我尝试过输入多种类型的数组和数组列表
错误: 原因:java.lang.IllegalArgumentException:无法在 FLOAT32 类型的 TensorFlowLite 张量和 java.lang.String 类型的 Java 对象(与 TensorFlowLite 类型 STRING 兼容)之间进行转换。
model.add(layers.Dense(500, input_dim=3, activation='relu'))
model.add(layers.Dense(1, activation= "relu"))
model.summary() #Print model Summary
model.compile(loss='mean_squared_error',optimizer='adam')
model.fit(X_train,Y_train,epochs=1000,validation_split=0.3)
我如何转换:-
from tensorflow.contrib import lite
converter = lite.TFLiteConverter.from_keras_model_file( 'Model.h5')
tfmodel = converter.convert()
open ("model.tflite" , "wb") .write(tfmodel)
Android 的实现
ArrayList<String> list = new ArrayList<>();
list.add("-0.5698444");
list.add("-0.57369368");
list.add("-1.31490297");
try (Interpreter interpreter = new Interpreter(mappedByteBuffer)) {
interpreter.run(list, "output");
}
private MappedByteBuffer loadModelFile() throws IOException {
AssetFileDescriptor fileDescriptor = getAssets().openFd("model.tflite");
FileInputStream inputStream = new FileInputStream(fileDescriptor.getFileDescriptor());
FileChannel fileChannel = inputStream.getChannel();
long startOffset = fileDescriptor.getStartOffset();
long declaredLength = fileDescriptor.getDeclaredLength();
return fileChannel.map(FileChannel.MapMode.READ_ONLY, startOffset, declaredLength);
}
最佳答案
发现错误了。在行中,
try (Interpreter interpreter = new Interpreter(mappedByteBuffer)) {
interpreter.run(list, "output");
}
interpreter.run()
的第二个参数必须是 float[]
而不是 “output”
。模型运行时,float[]
将填充类概率。
向 interpreter.run()
方法提供输入和输出的正确方法:
Interpreter interpreter = new Interpreter(mappedByteBuffer)
float[][] inputs = new float[1][num_features]
// populate the inputs float array above
float[][] outputs = new float[1][num_classes]
interpreter.run( inputs , outputs )
float[] classProb = outputs[0]
关于java - tensorflow float32的转换与java对象不兼容,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56323246/