Обученная CNN модель не сохранена должным образом

Question

Я новичок в использовании тензорного потока, и я построил CNN, который обучен и протестирован с точностью 80-85%. Я попытался сохранить свою обученную модель, используя model.save('example.h5'), и загрузить файл, используя files.download('example.h5')

. После этого я попытался загрузить его на свой flask сервер, используя model = tf.keras.models.load_model('example.h5').

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

Answer 1

Распространенной причиной этого может быть то, что вы нормализуете свои данные во время обучения, но не перед прогнозированием. Убедитесь, что вы нормализовали входные данные, прежде чем запускать прогнозы, если вы тренировались на нормализованных данных.