在移动设备上实施MobilenetV2的问题

时间:2019-11-19 09:28:28

标签: android tensorflow mobilenet tf-lite

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

1 个答案:

答案 0 :(得分:0)

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