使用keras加載模型時如何解決錯誤

Question

我有一個燒瓶應用程序，我正在使用 tensorflow 和 keras 來加載我將要使用的模型。下面是一個代碼片段

from tensorflow.keras.models import Model , load_model
from keras.preprocessing import image

# Flask utils
from flask import Flask, redirect, url_for, request, render_template
from werkzeug.utils import secure_filename
from gevent.pywsgi import WSGIServer

app = Flask(__name__)


Model= load_model('models/model_pro1.h5') 

所以我然后在heroku上部署flask應用程序並且它成功構建，但是當我檢查日志時我看到我得到了heroku日志的錯誤圖像我已經嘗試安裝tensorflow和keras而沒有在requirements.txt中的特定版本文件，但沒有得到解決。 我想知道我做錯了什么，並提前致謝。 我正在使用這個 github repo github repo 中的代碼

Answer 1

從 Python 3 開始，所有字符串默認都是 unicode。 所以我認為你的開發環境，你構建和訓練模型的地方運行的是 python2，而 heroku env 運行的是 python3。

Answer 2

我將給出一個保存模型然后加載它的實例

import tensorflow as tf
from tensorflow import keras
model.save(./abc.h5  # saves in h5 format)
model.save(./defg)#defg is a folder having pb file 

要加載保存的模型，

new_model = tf.keras.models.load_model('./abc.h5') #for loading h5 model

或者

new_model = tf.keras.models.load_model('defg/) # for loading pb model

我認為你的 load_model 語法是你犯的錯誤