android - 自定义模型 [MLKit] - FirebaseMLException:执行 Firebase ML 任务时发生内部错误
问题描述
我想使用 MLKit 在 Android 中使用自定义 ML 模型,并且使用本地或远程模型都不起作用。因为我有一个 FirebaseMLException,所以我在这里专注于本地。
我试图一步一步地遵循官方文档。
我相信错误在于输入/输出格式,但我无法弄清楚它是什么。具有143
由整数标记的类。这是我构建输入/输出的方式:
inputOutputOptions = FirebaseModelInputOutputOptions.Builder()
.setInputFormat(
0,
FirebaseModelDataType.FLOAT32,
intArrayOf(
DIM_BATCH_SIZE,
DIM_IMG_SIZE_X,
DIM_IMG_SIZE_Y,
DIM_PIXEL_SIZE
)
)
.setOutputFormat(
0,
FirebaseModelDataType.INT32,
intArrayOf(1, 143)
)
.build()
在孔堆栈跟踪下方找到:
W/System.err: com.google.firebase.ml.common.FirebaseMLException: Internal error has occurred when executing Firebase ML tasks
W/System.err: at com.google.firebase.ml.common.internal.zze.zza(com.google.firebase:firebase-ml-common@@20.0.1:38)
at com.google.firebase.ml.common.internal.zzh.run(Unknown Source:4)
W/System.err: at android.os.Handler.handleCallback(Handler.java:873)
W/System.err: at android.os.Handler.dispatchMessage(Handler.java:99)
at com.google.android.gms.internal.firebase_ml.zzf.dispatchMessage(com.google.firebase:firebase-ml-common@@20.0.1:6)
W/System.err: at android.os.Looper.loop(Looper.java:193)
at android.os.HandlerThread.run(HandlerThread.java:65)
W/System.err: Caused by: java.lang.IllegalArgumentException: Cannot convert between a TensorFlowLite tensor with type FLOAT32 and a Java object of type [[I (which is compatible with the TensorFlowLite type INT32).
W/System.err: at org.tensorflow.lite.Tensor.throwExceptionIfTypeIsIncompatible(Tensor.java:233)
W/System.err: at org.tensorflow.lite.Tensor.copyTo(Tensor.java:116)
at org.tensorflow.lite.NativeInterpreterWrapper.run(NativeInterpreterWrapper.java:157)
W/System.err: at org.tensorflow.lite.Interpreter.runForMultipleInputsOutputs(Interpreter.java:250)
at com.google.android.gms.internal.firebase_ml.zzpz.runForMultipleInputsOutputs(com.google.firebase:firebase-ml-model-interpreter@@20.0.1:4)
W/System.err: at com.google.android.gms.internal.firebase_ml.zzpu.zza(com.google.firebase:firebase-ml-model-interpreter@@20.0.1:85)
W/System.err: at com.google.android.gms.internal.firebase_ml.zzpu.zza(com.google.firebase:firebase-ml-model-interpreter@@20.0.1:145)
at com.google.firebase.ml.common.internal.zzi.zza(com.google.firebase:firebase-ml-common@@20.0.1:33)
at com.google.firebase.ml.common.internal.zzk.call(Unknown Source:8)
W/System.err: at com.google.firebase.ml.common.internal.zze.zza(com.google.firebase:firebase-ml-common@@20.0.1:32)
... 6 more
我不知道如何解释这一行:
Cannot convert between a TensorFlowLite tensor with type FLOAT32 and a Java object of type [[I (which is compatible with the TensorFlowLite type INT32)
解决方案
我发现问题出在哪里,面对 MLKit 等待一组概率作为输出,这意味着一个范围内的一组float
值。[0.0, 1.0]
这意味着这个片段(包含错误的行):
.setOutputFormat(
0,
FirebaseModelDataType.INT32, // Wrong line
intArrayOf(1, 143)
)
应该用这个替换:
.setOutputFormat(
0,
FirebaseModelDataType.FLOAT32, // <== Expected array type is a float
intArrayOf(1, 143)
)
推荐阅读
- node.js - 如何使用 sqlite3 在 node.js 中更新,如 mongo 上的函数 findByIdAndUpdate?
- git - 什么样的 Git 命令序列可能会导致这个提交图
- python - 训练有效,但预测产生恒定值(使用 pytorch 的 cnn)
- typescript - 在 Firestore 中获取所有可以访问其子项的文档
- javascript - 在按钮单击中显示组件(使用挂钩)
- python - 为 Spotify API PKCE 创建代码挑战,为什么它不是字节形式?
- java - 从 c++ 将动态 java 代理加载到 jvm
- angular - 不能使用 Angular Material 日期范围日期过滤器
- python - TypeError:'str'和'int'的实例之间不支持'<='重复
- sql - 将 SQL 中的 VARCHAR 列转换为日期时间格式?