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

时间:2018-12-05 14:13:13

标签: android tensorflow bytebuffer tensor tensorflow-lite

Tensorflow在以下链接中提供了演示:

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

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

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

请解决该解决方案,我们如何使用label.txtINPUT_SIZE之类的常量更改OUTPUT_SIZE文件,以解决此错误。

注意

应用程序可以正常工作,但是在更改label.txt文件(例如,在文件中添加或删除名称。

崩溃:

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

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

功能请求: IOS演示中提供了更改标签文件大小和通过参数进行相应操作的操作
output_size = 1000;
此功能也应为Android版本。

谢谢。

1 个答案:

答案 0 :(得分:0)

作为功能请求,通常应将内容归档到Github。

如果您正在寻找本地修补程序,我想问题是该模型输出1x1000,您需要将实际模型修改为1x1001(及其之前的所有形状) )。