Python backend.get_session方法代碼示例

本文整理匯總了Python中tensorflow.python.keras.backend.get_session方法的典型用法代碼示例。如果您正苦於以下問題：Python backend.get_session方法的具體用法？Python backend.get_session怎麽用？Python backend.get_session使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類tensorflow.python.keras.backend的用法示例。

在下文中一共展示了backend.get_session方法的3個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: save_model_to_tensorflow

# 需要導入模塊: from tensorflow.python.keras import backend [as 別名]
# 或者: from tensorflow.python.keras.backend import get_session [as 別名]
def save_model_to_tensorflow(self, new_model_folder, new_model_name=""):

        """
        'save_model_to_tensorflow' function allows you to save your loaded Keras (.h5) model and save it to the Tensorflow (.pb) model format.
        - new_model_folder (required), the path to the folder you want the converted Tensorflow model to be saved
        - new_model_name (required), the desired filename for your converted Tensorflow model e.g 'my_new_model.pb'

        :param new_model_folder:
        :param new_model_name:
        :return:
        """

        if(self.__modelLoaded == True):
            out_prefix = "output_"
            output_dir = new_model_folder
            if os.path.exists(output_dir) == False:
                os.mkdir(output_dir)
            model_name = os.path.join(output_dir, new_model_name)

            keras_model = self.__model_collection[0]


            out_nodes = []

            for i in range(len(keras_model.outputs)):
                out_nodes.append(out_prefix + str(i + 1))
                tf.identity(keras_model.output[i], out_prefix + str(i + 1))

            sess = K.get_session()

            from tensorflow.python.framework import graph_util, graph_io

            init_graph = sess.graph.as_graph_def()

            main_graph = graph_util.convert_variables_to_constants(sess, init_graph, out_nodes)

            graph_io.write_graph(main_graph, output_dir, name=model_name, as_text=False)
            print("Tensorflow Model Saved")

開發者ID:OlafenwaMoses，項目名稱:ImageAI，代碼行數:40，代碼來源:__init__.py

示例2: _init_session

# 需要導入模塊: from tensorflow.python.keras import backend [as 別名]
# 或者: from tensorflow.python.keras.backend import get_session [as 別名]
def _init_session():
    from tensorflow.python.keras import backend
    sess = backend.get_session()
    tf.get_default_graph()
    set_session(sess)
    return sess

開發者ID:FederatedAI，項目名稱:FATE，代碼行數:8，代碼來源:dense_model.py

示例3: on_train_begin

# 需要導入模塊: from tensorflow.python.keras import backend [as 別名]
# 或者: from tensorflow.python.keras.backend import get_session [as 別名]
def on_train_begin(self, logs=None):
        K.get_session().run(tf.tables_initializer())

開發者ID:marco-willi，項目名稱:camera-trap-classifier，代碼行數:4，代碼來源:hooks.py

注：本文中的tensorflow.python.keras.backend.get_session方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台，相關代碼片段篩選自各路編程大神貢獻的開源項目，源碼版權歸原作者所有，傳播和使用請參考對應項目的License；未經允許，請勿轉載。