android - tensorflow lite 问题：输出标签文件大小是固定的？？android中的输出或输入张量维度不匹配问题

Question

Tensorflow 可以在以下链接中获得演示：

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/lite/java/demo

请通过添加或删除 text(class) 来更改上述文件中的 labels.txt 文件，这会使应用程序崩溃。

它使给定的解决方案无法更改输出的大小。

请解决解决方案，我们如何label.txt使用一些常量更改文件，例如INPUT_SIZE解决OUTPUT_SIZE这个错误。

笔记

应用程序按原样工作正常，但在更改label.txt文件时崩溃，例如在文件中添加或删除名称。

碰撞 ：

无法在TensorFlowLite形状为 [1, 1001] 的张量和形状为 [1, 1000] 的 Java 对象之间复制。

****以上演示是使用已经存在的gradle配置设置构建的。**

功能请求： 通过参数更改标签文件大小和相应的工作在IOS演示中可用
output_size = 1000;
此功能也应在 Android 版本中。

谢谢。

标签： androidtensorflowbytebuffertensortensorflow-lite

Answer 1

作为功​​能请求，您通常应该向 Github 提交内容。

如果您正在寻找本地修复，我猜问题是模型输出 1x1000，您需要将实际模型修改为 1x1001（以及它之前的所有形状）。