Как загрузить ввод в (.pb) обученную модель? - PullRequest
Новая
       35

Как загрузить ввод в (.pb) обученную модель?

0 голосов
/ 06 июля 2019

На самом деле я работаю над приложением для определения эмоций на лице.Я обучил модель на наборе данных FER2013 с использованием керас.Поэтому для предсказания эмоций по лицу я делаю это 

from keras.preprocessing import image

from keras.preprocessing.image import ImageDataGenerator
import numpy as np
import matplotlib.pyplot as plt


file = 'photo.jpg'
true_image = image.load_img(file)
img = image.load_img(file, grayscale=True, target_size=(48, 48))

x = image.img_to_array(img)
x = np.expand_dims(x, axis = 0)

x /= 255

custom = model.predict(x)
emotion_analysis(custom[0])

x = np.array(x, 'float32')
x = x.reshape([48, 48]);

plt.gray()
plt.imshow(true_image)
plt.show()

После этого я сохранил свою модель в виде файла .pb.Теперь я не знаю, как загрузить эту модель и предсказать эмоцию из файла .jpg.Обученные изображения были в оттенках серого 48X48Пожалуйста, помогите мне.

...