tensorflow kears转pb

咚咚咚何

于 2022-06-09 17:42:23 发布

阅读量208

点赞数

分类专栏： TensorFlow 文章标签： tensorflow python 人工智能 tensorflow2.0 keras

本文链接：https://blog.youkuaiyun.com/qq_32435729/article/details/125208048

版权

TensorFlow 专栏收录该内容

4 篇文章

订阅专栏

        model = tf.keras.models.Model()
        function = _saving_utils.trace_model_call(model)
        concrete_func = function.get_concrete_function()
        frozen_func = convert_variables_to_constants_v2(
            concrete_func, lower_control_flow=False)
        input_shape = {"input_1" : [1，640，640，3]}
        _set_tensor_shapes(frozen_func.inputs, input_shape)
        frozen_func.graph.as_graph_def()
        tf.io.write_graph(graph_or_graph_def=frozen_func.graph,
                        logdir="./frozen_models",
                        name="model.pb",
                        as_text=False)