首页 > 解决方案 > MobilenetV2 在移动设备上实现的问题

问题描述

在 cifar10 数据集上训练的 MobilenetV2 tflite 模型需要形状为 1x32x32x3 的输入图像(因为在训练期间使用了批处理标志)但是为了将模型转储到移动设备中,我从相机中获得了 32x32x3 形状图像,因此现在无法将数据输入 tflite.interpreter .......任何建议

标签: androidtensorflowmobilenettensorflow-lite

解决方案


1x32x32x3 矩阵与 32x32x3 矩阵具有相同数量的元素,因此您可以将图像中的数据馈送到 tflite 解释器的输入张量,而不会导致任何错误。


推荐阅读