向tensorflow冻结图pb添加元数据

2024-04-25 21:53:10 发布

您现在位置:Python中文网/ 问答频道 /正文

为了共享经过训练的tensorflow网络,我们将图冻结到.pb文件中。我们还创建了一个xml文件,其中包含一些元数据,如输入张量和输出张量、要应用的预处理类型、训练数据信息等。然后使用Java或C#通过加载图形和计算张量等为模型提供服务

为了使共享更容易,我想在.pb文件的某处包含这个xml数据。有什么办法吗?一个想法是把它作为tf.常数,但我不知道如何将它连接到正规图。你知道吗

注意这是使用freeze_graph.py。新的SavedModel格式更合适吗?你知道吗


Tags: 文件数据模型网络信息图形类型tf