Как использовать изображение в качестве входа для модели TFLite? - PullRequest
1 голос
/ 28 марта 2019

Я обучил модель в Керасе, затем сохранил ее и преобразовал в файл TFLite. Чтобы проверить мою модель Keras, я использовал следующий код для преобразования изображения в массивы:

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)

, который помещает его в массив 4d формы (1, 224, 224, 1). У меня вопрос, как подготовить данные таким же образом для модели TFLite на Android? И требует ли файл TFLite данные в том же формате?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...