Сначала я работаю в django -обстановке.
Я хочу загрузить модель keras из settings.py и передать модель в function.py, чтобы я мог использовать модель в functions.py.
, но я получаю следующую ошибку :
InvalidArgumentError at /camera2/
Tensor input_1:0, specified in either feed_devices or fetch_devices was not found in the Graph
Некоторые люди отвечают на этот тип ошибки, чтобы использовать этот код:
from tensorflow import Graph, Session
import tensorflow as tf
import keras
thread_graph = Graph()
with thread_graph.as_default():
thread_session = Session()
with thread_session.as_default():
model = keras.models.load_model(path_to_model)
graph = tf.get_default_graph()
и
def _load_model_from_path(path):
global gGraph
global gModel
gGraph = tf.get_default_graph()
gModel = load_model(path)
Но это не сработало для меня и Я думаю, что ответ работает для загрузки модели в многопоточном случае.
Я не знаю, как исправить эту ошибку
Мой код ниже
###################################################
# settings.py/the file where to load the models
###################################################
from keras.models import load_model
...
MODEL_ROOT = os.path.join(BASE_DIR, 'yeom')
GMODEL = load_model(MODEL_ROOT +'/MobileNetV2(full).h5')
####################################################################
# function.py/the file to which settings.py passed the keras model
####################################################################
from django.conf import settings
...
model = settings.GMODEL
pred_y = model.predict(image_data)
...
.. . опущена секция кода