tensorflow-lite出现以下错误:#vk为输出索引0获取了1个类,根据标签Map,应为2

4si2a6ki  于 2021-07-03  发布在  Java
关注(0)|答案(1)|浏览(308)

在将元数据添加到我的tflite文件以在ml工具包中使用它之后,我得到了一个错误 Calculator::Open() for node "ClassifierClientCalculator" failed: #vk Got 1 class(es) for output index 0, expected 2 according to the label map. 我编辑了元数据中的类数以及标签文件中的类数,但仍然得到相同的错误。我的模型是一个二值图像分类模型,所以首先应该有两个类。

f0brbegy

f0brbegy1#

基于https://developers.google.com/ml-kit/custom-models#model-兼容性,如果输出包含两个类,则输出应为(12)或(1112)。你能再检查一下你的输出层吗?

相关问题