[英]tensorflow lite issue : output labels file size is fixed ?? output or input tensor dimension mismatch issue in android
Tensorflow在以下鏈接中提供了演示:
https://github.com/tensorflow/tensorflow/tree/master/tensorflow/lite/java/demo
請通過添加或刪除text(class)來更改上述文件中的labels.txt文件,這會使應用程序崩潰。
這使得給定的解決方案無法更改輸出的大小。
請解決該問題,我們如何使用一些常量(例如INPUT_SIZE
或OUTPUT_SIZE
更改label.txt
文件,從而解決了該錯誤。
注意
應用程序可以正常運行,但是在更改label.txt
文件時會崩潰,例如,在文件中添加或刪除名稱。
崩潰:
無法在形狀為[1,1001]的TensorFlowLite
張量與形狀為[1,1000]的Java對象之間進行復制。
****以上演示是使用已經存在的gradle
配置設置構建的。**
功能要求:更改標簽文件的大小並通過IOS
演示中的參數進行相應的操作
output_size = 1000;
此功能也應為Android版本。
謝謝。
作為功能請求,通常應將內容歸檔到Github。
如果您正在尋找本地修復程序,那么我想問題是模型輸出1x1000,您需要將實際模型修改為1x1001(及其之前的所有形狀)。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.