如果您的 Android 应用程序使用 TensorFlow Lite 模型作为音频分类器,并且使用您自己的自定义转换的 tflite 模型导致应用程序崩溃,请尝试以下步骤解决问题:
import tensorflow as tf
# Load the model
interpreter = tf.lite.Interpreter(model_path="custom_model.tflite")
# Check if the model is valid
interpreter.allocate_tensors()
// Load the model
MlModel mlModel = new MlModel(getApplicationContext(), "custom_model.tflite");
// Initialize the executor
Interpreter tflite = mlModel.createInterpreter();
tflite_convert --graph_def_file=custom_model.pb \
--output_file=custom_model.tflite \
--output_format=TFLITE \
--input_shape=1,16000,1 \
--input_array=input \
--output_array=output \
--inference_type=FLOAT \
--mean_values=0 \
--std_dev_values=1
通过以上步骤,您可以尝试修复自定义转换 tflite 模型导致 AudioClassifier 崩