“Prediksi kesalahan prasyarat tensorflow” Kode Jawaban

Prediksi kesalahan prasyarat tensorflow

config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True

first_graph = tf.Graph()
first_session = tf.compat.v1.Session(config=config)

second_graph = tf.Graph()
second_session = tf.compat.v1.Session(config=config)

first_model_path = 'first_model.h5'
second_model_path = 'second_model.hdf5'

with first_graph.as_default(), first_session.as_default():
    first_prediction(image,first_model_path)

#second model loading
with second_graph.as_default(), second_session.as_default():
    second_model = load_model(second_model_path)
    second_prediction(image,second_model)