首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

将.pb从Tensorflow 1.14迁移到2.0

是指将Tensorflow 1.14版本中使用的.pb文件(Protocol Buffer文件)迁移到Tensorflow 2.0版本中使用的新格式。

Tensorflow 2.0是Tensorflow的一个重大更新版本,引入了许多新特性和改进。在Tensorflow 2.0中,推荐使用SavedModel格式代替之前的.pb文件。SavedModel是一种更加灵活和可扩展的模型保存格式,可以更好地支持模型的版本控制、模型的部署和模型的迁移。

下面是将.pb从Tensorflow 1.14迁移到2.0的步骤:

  1. 确保已经安装了Tensorflow 2.0版本。可以使用以下命令安装最新版本的Tensorflow:
代码语言:txt
复制
pip install tensorflow
  1. 使用Tensorflow 1.x版本的tf.compat.v1模块加载.pb文件,并将其转换为SavedModel格式。可以使用以下代码:
代码语言:txt
复制
import tensorflow.compat.v1 as tf

# 加载.pb文件
graph_def = tf.GraphDef()
with tf.gfile.GFile('model.pb', 'rb') as f:
    graph_def.ParseFromString(f.read())

# 转换为SavedModel格式
with tf.Session() as sess:
    tf.import_graph_def(graph_def, name='')
    builder = tf.saved_model.builder.SavedModelBuilder('saved_model')
    builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING])
    builder.save()
  1. 迁移完成后,可以使用Tensorflow 2.0的tf.saved_model.load函数加载SavedModel格式的模型,并进行后续的操作。可以使用以下代码:
代码语言:txt
复制
import tensorflow as tf

# 加载SavedModel格式的模型
model = tf.saved_model.load('saved_model')

# 进行后续的操作
...

需要注意的是,迁移过程中可能会遇到一些兼容性问题,例如Tensorflow 2.0中某些API的参数名称或用法可能与Tensorflow 1.x不同。在迁移过程中,可以参考Tensorflow官方文档和API文档,查找相关的变化和更新。

推荐的腾讯云相关产品和产品介绍链接地址:

  • 腾讯云TensorFlow:https://cloud.tencent.com/product/tensorflow
  • 腾讯云AI引擎:https://cloud.tencent.com/product/aiengine
  • 腾讯云机器学习平台:https://cloud.tencent.com/product/tiia
  • 腾讯云GPU服务器:https://cloud.tencent.com/product/cvm_gpu
  • 腾讯云弹性容器实例:https://cloud.tencent.com/product/eci
页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

  • tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例

    在tf1.x的环境下, tf2.0保存的weights转为pb模型: 如果在tf2.0下保存的模型符合上述的三个定义, 那么这个.h5文件在1.x环境下其实是可以直接用的, 因为都是通过tf.keras...", as_text=False) 运行成功后, 会在当前目录下生成一个model文件夹, 里面有生成的tf_model.pb文件, 至此, 我们就完成了tf2.0下训练的模型转到tf1.x下的pb模型...这个转换的重点就是通过keras这个中间商来完成, 所以我们定义的模型就必须要满足这个中间商定义的条件 补充知识:tensorflow2.0降级及如何别的版本升到2.0 代码实践《tensorflow...实战GOOGLE深度学习框架》时,由于本机安装的tensorflow2.0版本与配套书籍代码1.4的API不兼容,只得tensorflow降级为1.4.0版本使用,降级方法如下 1 pip uninstall...二 别的版本升级到2.0 自动卸载与其相关包 pip uninstall tensorflow 安装某版本 pip install –no-cache-dir tensorflow==x.xx

    2.5K20

    TensorFlow 2.0 beta版本来了!

    并且,在TensorFlow 1.14版本中提供了的v2兼容性模块,增加了Keras那样的2.0特性支持,如模型子类化,简化了自定义训练循环的API,为大多数类型的硬件添加了分发策略支持等等。...虽然可以通过TensorFlow团队提动的脚本1.0的代码自动转换为2.0的代码,而2.0 API中也有一个v1子模块提供对老API的支持,但1.0的代码不能直接在TensorFlow 2.0上运行,...不知道TensorFlow 1.0和2.0会不会像python 2和python 3那样共存很多年。 你会在项目中使用TensorFlow 2.0么?...在使用TensorFlow 2.0的过程中,有没有碰到什么问题,欢迎交流!...你还可以看: 尝鲜TensorFlow 2.0 [译]高效的TensorFlow 2.0:应用最佳实践以及有什么变化 [译]标准化Keras:TensorFlow 2.0中的高级API指南

    70820

    OpenVINO运行Tensorflow模型

    research/slim/nets/mobilenet_v1.md下载MobileNet_v1_1.0_224模型,解压到目录E:\model后,对mobilenet_v1_1.0_224_frozen.pb...scale_values [127.5,127.5,127.5] 参数介绍: --input_model :指定输入模型路径 --input_shape :指定模型的输入Tensor的shape,如果不指定,则会自动pb...中读取 --output :指定输出节点名称,如果不指定,会自动图中提取。...注意,如果转换过程中出错了,可以尝试卸载Tenorflow,可能是因为Tensorflow版本问题,改为Tensorflow1.14-cpu版本,笔者这边使用1.14-cpu版本没有问题。...注意:如果懒得配置,可以附件中下载笔者已经搭建好的环境,可直接用VS2017打开运行 2.2 代码编写 E:\model拷贝到项目根目录,输入以下代码。

    2.5K60

    『带你学AI』极简安装TensorFlow2.x的CPU与GPU版本教程

    我们再次进入 conda activate TF_2C ,便于执行下述命令 1.1.2 安装TF2.0 CPU版本(后面的 -i 表示国内清华源下载,速度比默认源快很多) pip install tensorflow...GPU版本(后面的 -i 表示国内清华源下载,速度比默认源快很多) pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn... TensorFlow 2.1 开始,pip 包 tensorflow 即同时包含 GPU 支持,无需通过特定的 pip 包 tensorflow-gpu 安装 GPU 版本。...TensorFlow2.0 gpu版本安装 conda install cudatoolkit=10.0 cudnn=7 pip install tensorflow-gpu==2.0 TensorFlow1.15.../1.14/1.13.1 gpu版本安装 conda install cudatoolkit=10.0 cudnn=7 pip install tensorflow-gpu==1.15 1.4 最后我们测试一个使用

    2.5K10

    如何PyTorch、TensorFlow模型转换为PaddlePaddle模型

    本文手把手教你使用X2PaddlePyTorch、TensorFlow模型转换为PaddlePaddle模型,并提供了PaddlePaddle模型的使用实例。...TensorFlow模型转换 为PaddlePaddle模型 注:model.pb为TF训练好的模型,pb_model为转换为PaddlePaddle之后的文件。 1....安装依赖库: 在实践下述代码前,你需要确保本地环境满足以下依赖库: TensorFlow1.14 PaddlePaddle1.8 pip install paddlepaddle -i https://...在终端中运行如下命令安装TF1.14: pip install tensorflow==1.14 –i https://mirror.baidu.com/pypi/simple 执行如下命令开启训练过程...python work/X2Paddle_ISSUE/train.py 在本地终端输入以下代码TF模型转换为PaddlePaddle模型: x2paddle --framework=tensorflow

    2.6K20

    使用TensorFlow的经验分享

    (Bypass)设置,既大幅度减少了网络的参数量,又在一定程度上缓解了梯度消失问题的产生 五、学习Tensorflow1.0与tensorflow2.0 1....什么是tensorflow 一个核心开源库,可以帮助您开发和训练机器学习模型。,Tf1.0主要采用静态计算图,tf2.0主要采用动态计算图 2....哪个开始学习 个人建议直接tf2.0学起,2.0的很多代码都比较简洁,比如训练模型,保存模型,加载模型之类的非常方便,上手比较快。...遇到1.0的代码可以用2.0自带的代码转换,1.0代码转为2.0代码使用。 3.学习tf中的各种库 1....解决办法: Path的路径转为str即可。 问题九:pb文件保存后加载问题 出现原因: 在模型训练结束后,我打算h5文件转为pb文件,进行模型的部署,转换后我打算加载pb文件测试是否能使用。

    1.4K12

    干货 | 英特尔神经网络计算棒实现对象检测加速推理

    物体检测模型的加速,涉及到的内容有tensorflow物体检测模型,OpencvDNN模块的使用,OpenVINO的使用。...tensorflow物体检测模型 Google开源的目标检测 API包含了许多优秀的检测模型,可任君选用,地址为: https://github.com/tensorflow/models/blob/master...OpencvDNN调用 OpencvDNN模块调用tensorflow物体检测模型,需要通过.config及.pb文件生成一个.pbtxt文件,进入opencv下的dnn目录下,本人目录为 D:\opencv...理论上,只需要将硬件CPU改为NCS2即可,: DNN_TARGET_CPU 改为 DNN_TARGET_MYRIAD 代码如下: Net face_detection_net = readNetFromTensorflow...其实只要用tensorflow物体检测框架训练,然后通过上述步骤,就可以模型迁移到自己的数据集上,如下图,是我用ssd_mobilenet_v2迁移到人头检测数据集的效果,跑在NCS2上的,效果如下图

    2.2K70

    如何内网使用Cloud云最新MSP迁移工具箱Docker镜像?

    镜像仓库:可以直接使用 docker pull 命令镜像仓库拉取。...工具箱包含功能 云步骤和方案 整个迁移过程按照不同阶段进行细化,在每一个具体的实施步骤中提供最佳实践和相应的工具。即使对云项目不熟悉,也可以跟随云步骤的引导一步步完成云实施。...大大缩减资源梳理工作量的同时,可以分析的结果与迁移成本估算、进度规划等模块衔接,为迁移过程提供更多的支持。 产品选型 友商云迁移到腾讯云,会面临同一类型产品名称不同、规格不同的困难。...产品选型对照表友商云与腾讯云常用对标产品的名称/缩写以及具体的规格型号进行一一对应,解决迁移选型的困难。...redirect=15& 负载均衡实现跨地域绑定2.0:https://cloud.tencent.com/act/cps/redirect?

    3.3K41

    如何使用 TensorFlow mobile PyTorch 和 Keras 模型部署到移动设备

    安卓手机的迅速普及很大程度上是因为各式各样的智能 app,地图到图片编辑器应有尽有。随着深度学习的出现,我们的手机 app 变得更加智能。...在这篇文章中,我阐释如何使用 TensorFlow mobile PyTorch 和 Keras 部署到移动设备。... Keras 转成 TensorFlow 模式 到这一步,你已经有了 Keras 模式,无论是 PyTorch 转化而来的还是直接用 Keras 训练而获得的。... TensorFlow Mobile 添加到你的项目中 TensorFlow 有 2 个针对移动设备的库,分别是「TensorFlow Mobile」和「TensorFlow Lite.」Lite 版本设计得非常小...为了得到更新奇的体验,你的 App 应当安卓文件系统加载图像或用摄像头抓取图像,而不是资源文件夹加载。 总结 移动端的深度学习框架最终转变我们开发和使用 app 的方式。

    3.6K30

    如何使用TensorFlow mobile部署模型到移动设备

    安卓手机的迅速普及很大程度上是因为各式各样的智能 app,地图到图片编辑器应有尽有。随着深度学习的出现,我们的手机 app 变得更加智能。...在这篇文章中,我阐释如何使用 TensorFlow mobile PyTorch 和 Keras 部署到移动设备。... Keras 转成 TensorFlow 模式 到这一步,你已经有了 Keras 模式,无论是 PyTorch 转化而来的还是直接用 Keras 训练而获得的。... TensorFlow Mobile 添加到你的项目中 TensorFlow 有 2 个针对移动设备的库,分别是「TensorFlow Mobile」和「TensorFlow Lite.」Lite 版本设计得非常小...为了得到更新奇的体验,你的 App 应当安卓文件系统加载图像或用摄像头抓取图像,而不是资源文件夹加载。 总结 移动端的深度学习框架最终转变我们开发和使用 app 的方式。

    1.1K50

    干货:TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理

    要搭建TensorFlow的GPU版本,首先需要的必备条件就是一块能够支持CUDA的NVIDIA显卡,因为在搭建TensorFlow的GPU版本时,首先需要做的一件事就是安装其基础支持平台CUDA和其机器学习库...其次还要了解一下不同的TensorFlow版本所需要对应安装的CUDA和cuDNN版本是多少,因为在TensorFlow的GPU版本安装过程中,如果对应的CUDA版本和cuDNN版本不正确的话,是无法正常使用...下表整理出了TensorFlow1.2到最新版本的CUDA和cuDNN所对应的版本集合。...1.11 CUDA Toolkit 9.0 cuDNN v7.1 1.12 CUDA Toolkit 9.0 cuDNN v7.3 1.13 CUDA Toolkit 10.0 cuDNN v7.3 1.14...CUDA Toolkit 10.0 cuDNN v7.4 1.15 CUDA Toolkit 10.0 cuDNN v7.5 2.0 CUDA Toolkit 10.0 cuDNN v7.6 2.1.0

    3.6K10

    Tensorflow模型导出为一个文件及接口设置

    有没有办法导出为一个pb文件,然后直接使用呢?答案是肯定的。在文章《Tensorflow加载预训练模型和保存模型》中提到,meta文件保存图结构,weights等参数保存在data文件中。...1 模型导出为一个文件 1.1 有代码并且从头开始训练 Tensorflow提供了工具函数tf.graph_util.convert_variables_to_constants()用于变量转为常量。...可以看到通过tf.graph_util.convert_variables_to_constants()函数变量转为了常量,并存储在graph.pb文件中,接下来看看如何使用这个模型。...那么,这4个文件导出为一个pb文件方法如下: import tensorflow as tf with tf.Session() as sess: #初始化变量 sess.run(tf.global_variables_initializer...) w2 = tf.Variable(30.0, name="w2") #这里b1改为placeholder,让用户输入,而不是写死 #b1= tf.Variable(2.0,name="bias"

    1.8K20

    教程 | 如何模型部署到安卓移动端,这里有一份简单教程

    安卓手机的迅速普及在很大程度上得益于各种各样的智能应用,地图到图片编辑器无所不有。随着深度学习技术的兴起,移动应用注定会变得更加智能。深度学习加持下的下一代移动应用专门为你学习和定制功能。...使用 TensorFlow Mobile 模型部署到安卓设备上包括三个步骤: 训练好的模型转换成 TensorFlow 格式; 向安卓应用添加 TensorFlow Mobile 依赖项; 编写相关的...在两个框架中都定义 Squeezenet,然后使用下面的方法 PyTorch 框架的权重迁移到 Keras 框架中。 创建一个 convert.py 文件,引入下面的代码,并且运行脚本。... Keras 模型转化为 TensorFlow 模型 此时,你已经有了一个 PyTorch 模型转换而来的 Keras 模型,或者直接使用 Keras 训练得到的模型。...(keras_model,output_dir=output_dir,model_name="squeezenet.pb") print("MODEL SAVED") 上述代码 squeezenet.pb

    3.3K10
    领券