Как кормить обученную Python модель TensorFlow пакетами? - PullRequest
0 голосов
/ 30 мая 2020

Я создал собственную модель и обучил ее с помощью ImageTrainGenerator - из Keras, используя flow_from_directory.

Примерно так: как обучать модель пакетами . Все работает нормально, сгенерированные партии проверил, картинки такие, какие должны быть.

Моя проблема в том, что я хочу использовать эту обученную модель в онлайн-распознавании лиц. Я обрезаю грани желаемой ширины и высоты, конвертирую в массив, но предсказание ужасное.

Я считаю, что изображение, передаваемое в реальном времени, должно быть таким же, как и то, что создает Imagetraingenerator (пакеты). Есть идеи, как преобразовать изображение cv2.imread(path) в пакет, чтобы предсказать класс?

1 Ответ

0 голосов
/ 30 мая 2020

Вам просто нужно добавить размеры партии, чтобы преобразовать ее в партию с 1 образцом: np.expand_dims(img, axis=0).

...