如何从TensorFlow .pb模型中获取权重格式？
从TensorFlow .pb模型中获取权重格式的方法如下：
- 首先，加载TensorFlow模型并创建一个会话（Session）：import tensorflow as tf # 加载模型 graph = tf.Graph() with tf.gfile.FastGFile('model.pb', 'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) with graph.as_default(): tf.import_graph_def(graph_def, name='') # 创建会话 sess = tf.Session(graph=graph)
- 获取权重变量的名称：# 获取所有的操作节点 operations = graph.get_operations() # 遍历操作节点，找到权重变量的名称 weight_names = [] for op in operations: for output in op.outputs: if output.dtype == tf.float32: weight_names.append(output.name)
- 根据权重变量的名称获取权重值：# 获取权重值 weights = [] for name in weight_names: weight = graph.get_tensor_by_name(name + ':0') weight_value = sess.run(weight) weights.append(weight_value)
通过以上步骤，你可以从TensorFlow .pb模型中获取权重格式。注意，这里的权重值是以NumPy数组的形式返回的，你可以根据需要进行进一步处理或使用。
推荐的腾讯云相关产品：腾讯云AI智能图像识别，产品介绍链接地址：https://cloud.tencent.com/product/ai_image
相关·内容
1回答
如何从TensorFlow .pb模型中获取权重格式？
c++、tensorflow、model
我想重组tensorflow .pb模型的节点，所以我首先从GraphDef获取NodeDef，并使用NodeDef.attr().for "Conv2D“的节点。我可以从attr获取诸如步幅、填充、data_format、use_cudnn_on_gpu等参数，但不能获取权重格式参数。我使用的语言是c++。如何获得它！谢谢!
浏览 18提问于2017-07-12得票数 8
1回答
如何从没有会话的keras .h5文件中获取Tensorflow会话
docker、tensorflow、keras、tensorflow-serving
参数设置为False，以便保存整个模型(而不仅仅是权重)。因此，我在问是否有可能在会话关闭后(例如，由于您的笔记本内核崩溃)，以某种方式从.h5权重文件中回溯提取TF会话。没有多少代码可以继续，但就是这样：
如何使用Tensorflow
浏览 1提问于2019-10-24得票数 1
1回答
在Tensorflow模型保存中，何时使用.ckpt vs .hdf5和.pb文件扩展名？
python、tensorflow、hdf5、tensorflow2.0、ckpt
Tensorflow解释说，模型：.ckpt或.hdf5或.pb。这里有很多文档，所以最好能更简单地比较什么时候使用哪种文件格式。为什么推荐它而不是.pb和.hdf5呢？它能节省空间吗？它是否包括其他文件格式不包含的数据？pb
另一方面，SavedModel格式除了参数值(检查点)外，还包括模型定义的计算的序列化描述。这种格式的模型独立于创建模型的源代码。因此，它们适合通
浏览 7提问于2020-01-23得票数 15
2回答
如何在Tensorflow的Java API中加载使用`saver.save`保存的模型
java、tensorflow
我正在尝试使用新的Java API从磁盘读取模型。
使用Tensorflow的Java API的展示了如何读取同时具有图形定义和参数权重的.pb模型文件。在Python端，Tensorflow建议使用Saver对象将模型保存到磁盘。它创建一个.meta文件，该文件具有该定义并具有用于权重的.data文件。在Python中，我使用new_saver=tf.train.import_meta_graph
浏览 5提问于2017-03-16得票数 2
2回答
将Azure自定义视觉训练模型与tensorflow.js结合使用
microsoft-cognitive、azure-cognitive-services、tensorflow.js
我已经用Azure Custom Vision训练了一个模型，并下载了适用于安卓的TensorFlow文件(参见：)。如何将其与一起使用根据我的研究，我还了解到还有不同的pb文件，但我找不到哪种类型的Azure Custom Vision导出。
我找到了。这是为了转换TensorFlow SavedModel (来自Azur
浏览 1提问于2018-04-15得票数 3
1回答
如何将Dlib权重转换为tflite格式？
tensorflow、dlib、tensorflow-lite
我想把.dat格式的人脸检测、人脸标记和人脸识别的Dlib权重转换成.tflite格式。Tensorflow lite要求输入格式为tensorflow_saved模型/冻结图(.pb)或keras (.h5)格式。将Dlib .dat转换为任何这些都是可行的。有人能帮我解决这个问题吗?
浏览 2提问于2019-01-07得票数 1
2回答
*.pb文件在TensorFlow中的用途是什么?它是如何工作的？
tensorflow
我正在使用一些实现来创建一个使用的人脸识别这个文件有什么用？我不确定它是如何工作的。控制台日志：distance = 0.72212267
代码的实际所有者的Github链接
浏览 0提问于2018-07-11得票数 38
回答已采纳
1回答
如何在Tensorflow中从.pb模型中获取权重
python、tensorflow
我训练了一个模型，然后通过冻结该模型创建了一个.pb文件。所以，我的问题是如何从.pb文件中获取权重，否则我必须做更多的处理来获取权重
@mrry，请指引我。
浏览 0提问于2017-09-09得票数 10
2回答
TensorFlow保存到文件中/从文件加载图形
python、tensorflow、protocol-buffers
根据我到目前为止收集到的信息，有几种不同的方法可以将TensorFlow图转储到文件中，然后将其加载到另一个程序中，但我还没有找到关于它们如何工作的明确示例/信息。()和tf.import_graph_def() ()将其加载回
冻结图形以将图和权重保存在一起()检查点文件是否可以加载到一个新的程序中<
浏览 4提问于2016-08-14得票数 106
回答已采纳
1回答
Tensorflow服务的保存模型
tensorflow、tensorflow-serving
我已经按照tensorflow服务指示在码头集装箱中服务模型。通过遵循的指示，我能够为mnist和初始模型服务。服务模型保存在以下结构中：|-- inception-export| |-- saved_model.pb|如何以格式转换输出，以便通过Tensorflow服务服务器提供经过再培训的模型？
如何为训练
浏览 3提问于2017-08-23得票数 0
1回答
如何将keras模型转换为协议缓冲区(.pb)文件？
python、tensorflow、keras、protocol-buffers
我有一个经过训练的keras模型，我想将它保存到一个协议缓冲区(.pb)文件中。当我这样做并加载模型时，预测是错误的(并且与原始模型不同)，权重也是错误的。下面是模型类型：> keras.engine.training.ModelK.set_learning_phase(0)
import <em
浏览 0提问于2019-08-10得票数 5
1回答
如何在tensorflow中创建训练模型的model.graphdef和config.pbtxt？
tensorflow、protocol-buffers
为了在triton推理服务器中提供模型，模型存储库应如下所示 <model-repository-path>/ config.pbtxt在TensorFlow中训练了一个模型之后， 1)How do you create model.graphdef?
浏览 65提问于2020-12-02得票数 0
4回答
一个已保存的Tensorflow .pb模型的权值恢复
python、tensorflow
我在这里见过很多关于恢复已经保存的TF模型的帖子，但是没有人能回答我的问题。Using TF 1.0.0
INCEPTION_LOG_DIR = '/tmp/inception
浏览 6提问于2017-07-27得票数 2
回答已采纳
1回答
加载Tensorflow模型文件未找到
python、path
我这样保存了我的tensorflow模型：
model.save('static/mysite/saved_model_weights/' + symbols[i] + ' saved model')model/saved_model.pb{saved_model.pbtxt|saved_model.pb}： OSError: SavedModel文件不存在于：/mysite/ saved _model_WAR权重<
浏览 1提问于2021-07-08得票数 0
1回答
如何将微小的yolov2模型转化为tensorflow模型？
android、tensorflow、tensorflow-lite、yolo、darkflow
我已经训练了一个定制的微型yolov2模型，并希望在应用程序中使用这个模型。我已经使用暗流将yolov2权重文件转换为.pb。转换后的模型在Xamarin Android应用程序中不起作用，出现了错误。这个说，.pb模型应该转换成tensorflow版本，在Android上运行，我也将模型转换为tensorflow模型，但是它仍然给出了错误
加载model.lite失败，不是有效
浏览 3提问于2019-07-29得票数 1
2回答
如何在tensorflow中结合variables.data和saved_model.pb
tensorflow、keras、tensorflow-serving、quantization
我是tensorflow和keras的新手。我使用keras训练了一个用于句子分类的CNN，并使用以下代码导出了该模型config = model.get_config() )我在variables文件夹和saved_model.pb我想在部署预测之前将这些文件合并到一
浏览 2提问于2017-11-28得票数 0
1回答
从TensorFlow图中提取权重以在Keras中使用
python、tensorflow、machine-learning、keras、deep-learning
我有一个Tensorflow图(output_graph.pb)，我是通过在之后在盗梦空间( Inception V3 )上进行迁移学习获得的。现在，我需要提取各个层的权重，以便将它们导入到用Keras创建的模型中。我知道从tensorflow直接将图转换为keras模型()是不可能的。
浏览 1提问于2018-09-30得票数 1
回答已采纳
2回答
如何将Tensorflow model.json转换为model.pb文件
tensorflow、machine-learning、computer-vision、tensorflow2.0、tensorflow.js
我正在尝试将model.json格式的张量流模型转换为model.pb格式。试图找到好的来源，但没有得到任何东西。我有model.json文件和二进制权重文件。最初，我对它们进行了转换，以便它可以在浏览器上工作，但现在需要它们作为.pb文件。请帮帮忙。https://www.tensorflow.org/js/tutorials/conversion/import_saved_model
浏览 101提问于2020-04-21得票数 2
2回答
用于珊瑚开发板(coral.ai)的.pb到tflite int8的转换问题
object-detection、tensorflow-lite、object-detection-api、google-coral、yolov4
我已经使用yolov4 darkenet Alexyab训练了目标检测模型，然后使用此存储库https://github.com/hunglc007/tensorflow-yolov4-tflite将生成的权重文件转换为tensorflow lite，我能够将yolo权重转换为pb和tflite float16，但使用此存储库转换为tflite int8时存在问题。我现在正在与珊瑚开发板，这需要tflite int8量化，请指导我如何</e
浏览 36提问于2021-04-07得票数 0
1回答
如何为tensorflow常量分配新值？
python、tensorflow、deep-learning
我正在从一个.pb文件加载一个TensorFlow模型。我想改变所有层的权重。我可以提取权重，但不能更改权重。我将graph_def模型转换为TensorFlow模型，但即使这样，我也无法为权重分配新值，因为权重存储在"Const“类型的张量中。
浏览 12提问于2019-09-03得票数 3
回答已采纳
扫码
添加站长 进交流群
领取专属 10元无门槛券
手把手带您无忧上云
相关资讯
热门标签
活动推荐
腾讯云开发者
