我在Keras中训练了一个模型,然后将其保存并转换为TFLite文件。为了测试我的Keras模型,我使用以下代码将图像转换为numpy数组:
img_path = r'path'
img = image.load_img(img_path, color_mode="grayscale", target_size=(img_height, img_width))
img = image.img_to_array(img)
img = numpy.expand_dims(img, axis=0)
img = img.reshape(1, img_width, img_height, 1)
将其放入形状为(1、224、224、1)的4d数组中。我的问题是如何为Android上的TFLite模型以相同的方式准备数据? TFLite文件是否要求数据格式相同?