向tensorflow冻结图pb添加元数据

向TensorFlow冻结图（Frozen Graph）PB添加元数据是指在TensorFlow模型中添加额外的元数据信息，以便更好地描述和管理模型。元数据可以包括模型的版本号、作者、创建日期、许可证信息等。

添加元数据可以通过以下步骤完成：

导入必要的库和模块：

import tensorflow as tf
from tensorflow.python.framework import graph_io

加载已经冻结的图：

input_graph = tf.GraphDef()
with tf.gfile.GFile('frozen_graph.pb', 'rb') as f:
    input_graph.ParseFromString(f.read())

创建一个新的图，并将已冻结的图复制到新图中：

output_graph = tf.GraphDef()
output_graph.CopyFrom(input_graph)

添加元数据到新图的node属性中：

metadata = {
    'version': '1.0',
    'author': 'Your Name',
    'created_date': '2022-01-01',
    'license': 'MIT License'
}
output_graph.node[0].metadata['metadata'] = tf.compat.as_bytes(str(metadata))

将新图保存到文件中：

output_graph_path = 'frozen_graph_with_metadata.pb'
with tf.gfile.GFile(output_graph_path, 'wb') as f:
    f.write(output_graph.SerializeToString())

在这个过程中，我们使用了tf.GraphDef类来表示图的定义，tf.gfile.GFile用于读取和写入图文件，graph_io模块用于复制图。通过修改新图的node属性，我们成功地向冻结图中添加了元数据。

添加元数据后，可以通过TensorFlow Serving、TensorFlow Lite等工具来加载和使用带有元数据的冻结图。

推荐的腾讯云相关产品：腾讯云AI智能图像处理（https://cloud.tencent.com/product/aiimage）提供了丰富的图像处理能力，可以与TensorFlow模型结合使用，实现更多的应用场景。

页面内容是否对你有帮助？

有帮助

没帮助

向tensorflow冻结图pb添加元数据

python、tensorflow

为了共享我们训练过的tensorflow网络，我们将图形冻结到一个.pb文件中。我们还创建了一个xml文件，其中包含一些元数据，如输入张量和输出张量、要应用的预处理类型、训练数据信息等。为了使共享更容易，我想在.pb文件中包含这个xml数据。有没有办法做到这一点？一种想法是将其作为tf.Constant，但我不知道如何将其与普通图形联系起来。

浏览 23提问于2019-02-12得票数 5

1回答

基于output_node_name的ssd移动网络v1 coco推理冻结模型

tensorflow、intel、ncsdk、movidius

我想把TensorFlow图编译成Movidius图。我使用了Model的ssd_mobilenet_v1_coco模型来在我自己的数据集上训练它。& saved_model/saved_model.pb 现在要将这个保存的模型转换为Movidius图。/tensorflow/tensorflow/python/tools/freeze_graph.py \

浏览 1提问于2019-07-17得票数 0

1回答

TensorFlowInferenceInterface返回低值

android、tensorflow、keras

我正在编写一个Android应用程序，它通过.pb文件使用Keras模型将叶子分类为健康或不健康。我使用TensorFlow与Android，特别是TensorFlowInferenceInterface类进行接口。input"，输入，1，宽度，高度，3)；inferenceInterface.run(新String[]{"activation_7/Sigmoid"})；当我使用Keras从木星笔记本运行.pb

浏览 1提问于2018-06-10得票数 0

回答已采纳

1回答

如何将Dlib权重转换为tflite格式？

tensorflow、dlib、tensorflow-lite

Tensorflow lite要求输入格式为tensorflow_saved模型/冻结图(.pb)或keras (.h5)格式。将Dlib .dat转换为任何这些都是可行的。有人能帮我解决这个问题吗?

浏览 2提问于2019-01-07得票数 1

1回答

将Tensorflow冷冻图转换为UFF进行TensorRT推理

python、tensorflow、tensorrt、uff

我想使用TensorRT (C++)的tensorflow模型，首先需要将.pb格式转换为.uff格式。执行此代码时： frozen_file="my_tf_model.pb", graphdef.Pa

浏览 22提问于2019-12-15得票数 2

回答已采纳

1回答

如何在Colab中将初始V4从.ckpt转换为.pb？

python、python-3.x、tensorflow、tensorflow-model-garden

为了将模型发送给他们，模型必须有扩展的.pb。所有模型都有扩展.ckpt，仅此而已。这里没有.meta或其他任何东西。我不知道如何转换为.pb。# Now let's download the pretrained model from tensorflow's model zoo. !wget http://download.tensorflow.org/m

浏览 2提问于2020-09-03得票数 1

回答已采纳

1回答

正确的pb文件将Tensorflow模型移动到ML.NET中

python、.net、tensorflow、nodes、ml.net

但是，ML.NET不接受h5格式，因此需要将其保存为pb。在查看在ML.NET中采用ML.NET的示例时，这个显示了一个TensorFlow模型，该模型以与格式类似的格式保存--由TensorFlow推荐(ML.NET 还建议“下载未冻结的SavedModel但是，当将pb文件保存并加载到Netron中时，我得到如下信息：再放大一点(在最右边)，正如你所看到的，它看起来一点也不应该。我使用来自的TensorFlow来确定输入/输出节点：当我试

浏览 9提问于2020-11-11得票数 8

回答已采纳

1回答

"TFlite“到Caffe模型

c++、tensorflow、opencv、caffe

我试图在Caffe模型(.caffemodel和.prototxt)中使用Opencv "cv::Net“，但我拥有的模型是".tflite”。谢谢。

浏览 8提问于2022-01-31得票数 1

1回答

如何从冻结的模型文件中获取tensorflow版本

tensorflow

我有一个tensorflow冻结图(.pb)文件。我如何得到确切的tensorflow版本，在模型上进行培训？

浏览 0提问于2019-06-12得票数 5

1回答

从keras中保存.pb和.pbtxt文件

c++、opencv、tensorflow、keras

我需要从.pb和.pbtxt文件中获取，以便使用经过训练的Keras模型和OpenCV在C++中的DNN模块。我可以通过执行.pb文件来获得model.save('model',save_format='tf')文件，但是我还没有找到获取.pbtxt文件的方法。

浏览 5提问于2020-06-09得票数 1

2回答

你如何识别输入和输出的名字，在张表图，像这个在图片附在这个帖子？

python、tensorflow、deep-learning、tensorboard、tensorflow-lite

我使用MobileNet_v1_1.0_224 tensorflow模型进行对象检测。现在，我有了我需要转换为tflite扩展的自定义冻结图(.pb文件)，这样我就可以在移动设备上使用我的模型了。我需要它们作为输入和输出参数，将冻结的图形(.pb文件)转换为tensorflow lite (.tflite)文件。

浏览 0提问于2018-09-15得票数 1

回答已采纳

1回答

运行Tensorflow模型而不安装Tensorflow

tensorflow、neural-network、tflearn

有没有一种方法可以在另一个系统上运行此模型，而无需在其上安装Tensorflow？它已经是预训练，所以我只需要运行数据通过它。我知道tf编译()，但设置起来似乎相当复杂。还有别的选择吗？

浏览 2提问于2018-01-25得票数 8

回答已采纳

2回答

将tensorflow模型打包为轮子文件

python、tensorflow、keras、setup.py、python-wheel

我已经创建了我的tensorflow模型，它将作为一个服务。代码将托管在客户端的本地服务器上。我不想给他们我的代码，但给他们一个轮子文件。但是在遵循了python的包分发之后，我的tensorflow文件就被破坏了。

浏览 0提问于2019-06-10得票数 0

2回答

TensorFlow -使用toco将.pb文件转换为.tflite时出错

android、tensorflow、tensorflow-lite

我使用下面的示例来创建tensorflow模型： --您可以从这里下载代码：也使用中的"2.冻结图形“部分来创建模型的*.pb文件。我尝试用toco命令行工具来转换*.pb文件，就像在上“将TensorFlow SavedModel转换为TensorFlow Lite”中描述的那样，并得到了以下错误： ( -c ) user@user-desktop:~/PycharmProjects/tensorflow_tutorial/<e

浏览 1提问于2018-05-12得票数 0

1回答

如何保存Tensorflow模型(不包含任何变量)以将其移植到OpenCV中

opencv、tensorflow、neural-network、opencv3.0、opencv3.1

我想知道保存我在python中训练过的tensorflow模型的正确方法是什么，这样我就可以使用opencv的dnn模块在OpenCV中导入它。这是我的Tensorflow图Y = tf.placeholder因此，我的问题是，如何将此图保存在Tensorflow中，以便可以使用cv::dnn::readNetFromTensorflow加载

浏览 3提问于2017-12-15得票数 2

回答已采纳

2回答

导出Tensorflow 2中的冻结图.pb文件

tensorflow、tensorflow2.0

我正在尝试Tensorflow 2 alpha，我一直试图冻结模型并将其导出到.pb图形文件中。在Tensorflow 1中，我可以这样做：frozen_graph_def = tf.graph_util.convert_variables_to_constants( sess.graph_def,with ope

浏览 0提问于2019-03-22得票数 7

回答已采纳

2回答

从字符串解析graph_def时出错

python、python-2.7、python-3.x、tensorflow、protocol-buffers

我正在尝试将Tensorflow图作为.pb文件运行一个非常简单的保存，但是当我将它解析回时有这个错误： File "test_import_stripped_bm.py这是我用来将它写到.pb的代码：import os

浏览 7提问于2017-10-03得票数 7

回答已采纳

2回答

如何优化推理一个简单、保存的TensorFlow 1.0.1图？

python、python-2.7、tensorflow

我无法在一个简单的保存的optimize_for_inference图(Python2.7；pip install tensorflow-gpu==1.0.1安装的包)上成功地运行TensorFlow模块背景下面是我的Python脚本，用于生成和保存一个简单的图形，以便将5添加到我的输入x placeholder操作中。图 out = ses

浏览 2提问于2017-07-28得票数 21

回答已采纳

1回答

冻结图的优化

python、tensorflow

我冻结了tensorflow图，并注意到实时使用该图形进行预测的速度太慢了。所以我想做图优化。 /python/tools：：没有这样的包'tensorflow/pytho

浏览 0提问于2017-07-19得票数 0

1回答

ValueError:保存Tensorflow检查点时没有要保存的变量

python、tensorflow

/usr/bin/pythonimport numpy as np with sess.as_default# Fails on this line: 'ValueError: No variables to save' saver.save(sess, '/path/to/tensorflow/graph_saver/

浏览 5提问于2017-02-09得票数 1

回答已采纳

点击加载更多