MLKit上的java Tensorflow Lite出现以下错误:#vk为输出索引0获取了1个类,根据标签映射,应为2个
在将元数据添加到我的tflite文件以在ML Kit中使用它之后,我得到了错误Calculator::Open() for node "ClassifierClientCalculator" failed: #vk Got 1 class(es) for output index 0, expected 2 according to the label map.
我编辑了元数据中的类数以及标签文件中的类数,但仍然得到了相同的错误。我的模型是一个二值图像分类模型,所以首先应该有两个类
# 1 楼答案
基于https://developers.google.com/ml-kit/custom-models#model-compatibility,如果输出包含两个类,则输出应该是(1*2)或(1*1*1*2)。你能再检查一下你的输出层吗