java - TensorBuffer：字节缓冲区的大小和形状不匹配

Question

我正在尝试在我的应用程序上显示我的张量缓冲区。它给了我这个错误，说大小和形状不匹配：

标签： java

Answer 1

首先我要问你，你在将模型转换为tfLite时是否使用了优化？如果是，那么您应该从这里阅读https://www.tensorflow.org/lite/performance/post_training_quantization。据说优化版本小了 4 倍，因此您需要将字节缓冲区创建为 [4 * 224 * 224 * 3] 以便它可以与 inputfeature0 匹配，我不知道您的输入是图像还是只是数据，如果它的图像比你应该先调整它的大小

val resize = Bitmap.createScaledBitmap(*input image*, 256, 256, true)

我根据我的经验将位图用于图像，希望对您有所帮助