как использовать модель keras, переданную из другого файла python в django - PullRequest
0 голосов
/ 04 мая 2020

Сначала я работаю в 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)


...


.. . опущена секция кода

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...