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

如何在Tensorflow保存pb文件之前访问文件夹名称?

在TensorFlow中,可以通过使用tf.train.Saver类来保存模型的pb文件。在保存pb文件之前,可以通过os.path.dirname()函数访问文件夹名称。

具体步骤如下:

  1. 导入必要的库:
代码语言:txt
复制
import tensorflow as tf
import os
  1. 定义模型和变量:
代码语言:txt
复制
# 定义模型
x = tf.placeholder(tf.float32, name='input')
y = tf.add(x, 1, name='output')

# 定义变量
var1 = tf.Variable(0, name='var1')
var2 = tf.Variable(1, name='var2')
  1. 创建Saver对象:
代码语言:txt
复制
saver = tf.train.Saver()
  1. 创建会话并保存模型:
代码语言:txt
复制
with tf.Session() as sess:
    # 初始化变量
    sess.run(tf.global_variables_initializer())

    # 保存模型
    save_path = saver.save(sess, os.path.join(os.path.dirname(__file__), 'model.pb'))
    print("Model saved in file: %s" % save_path)

在上述代码中,os.path.dirname(__file__)用于获取当前文件所在的文件夹路径,然后通过os.path.join()函数将文件夹路径和文件名拼接起来,最终得到保存pb文件的完整路径。

这样,就可以在保存pb文件之前访问文件夹名称,并将模型保存在指定的文件夹中。

注意:以上代码示例中的__file__是Python中的内置变量,表示当前脚本的文件名。如果代码不在脚本中运行,而是在交互式环境中运行,可以手动指定文件夹路径。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

TensorFlow 智能移动项目:1~5

运行这些示例的最简单方法是使用 Android Studio 在前面的文件夹中打开项目,“设置 Android Studio”部分中所示,然后通过编辑项目的build.gradle文件进行单个更改,并将...每个文件夹对应一个犬种,并且包含约 150 张图像(您无需为图像提供显式标签,因为文件夹名称用于标记文件夹中包含的图像): [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-BNHjmcvF...--output_graph表示再训练模型的名称和路径。 --output_labels是由图像数据集的文件夹(标签)名称组成的文件,稍后将其与经过重新训练的模型一起使用以对新图像进行分类。...创建一个新的assets文件夹,如图 2.13 所示,然后将您训练过的快速神经迁移模型从 iOS 应用中拖动(如果您在上一节中尝试过),或者从文件夹/tf_files中拖动,“训练快速神经样式迁移模型...创建一个新的资产文件夹,然后将在上一节的步骤 2 和 3 中生成的speech_commands_graph.pb和conv_actions_labels.txt文件拖放到assets文件夹中。

4.5K20

浅谈tensorflow模型保存pb的各种姿势

一,直接保存pb 1, 首先我们当然可以直接在tensorflow训练中直接保存pb为格式,保存pb的好处就是使用场景是实现创建模型与使用模型的解耦,使得创建模型与使用模型的解耦,使得前向推导inference...,并且可以把多个计算图保存到一个pb文件中,如果有多个MetaGraph,那么只会保留第一个MetaGraph的版本号。...,使用起来更加简单 这样和之前的导入pb模型一样,也是要知道tensor的name,那么如何在不知道tensor name的情况下使用呢,给add_meta_graph_and_variables方法传入第三个参数...checkpoint是检查点的文件文件保存了一个目录下所有的模型文件列表 model.ckpt.meta文件保存Tensorflow计算图的结果,可以理解为神经网络的网络结构,该文件可以被tf.train.import_meta_graph...,这个节点名称是原模型存在的结点,注意节点名称与张量名称的区别: :“input:0”是张量的名称,而“input”表示的是节点的名称 源码中通过graph = tf.get_default_graph

4.5K20
  • ensorFlow 智能移动项目:6~10

    之前我们已经重用了ios_image_load.*文件,此处tensorflow_utils.*文件主要用于加载映射的模型文件。...像以前一样创建一个新的assets文件夹,并将quickdraw_frozen_long_blacklist_strip_transformed.pb和classes.txt拖放到该文件夹​​中。...,这意味着您的应用的jniLibs文件夹中没有正确的 TensorFlow 本机库(图 7.9)。...您所见,在我们甚至开始训练模型之前一定要知道那条信息,这绝对是一件很高兴的事情,而这是我们从头开始构建的模型的好处。 开始训练过程。...只需创建一个名称为 AlphaZero 的新 Android Studio 应用,将alphazero19.pb模型文件复制到新创建的素材资源文件夹,然后将 compile 'org.tensorflow

    1.8K20

    TensorFlow学习笔记--自定义图像识别

    零、学习目标 本篇文章主要讲解自己的图像数据如何在TnesorFlow上训练,主要从数据准备、训练模型、验证准确率和导出模型并对图片分类。...在slim文件夹下新建 satellite 目录、satellite/data(训练和验证数据文件夹)、satellite/train_dir(保存训练日志和模型文件夹)、satellite/pretrained....ckpt文件复制到 satellite/pretrained 训练程序 在slim文件夹下启动命令行,输入如下命令开始训练(代码需要在TensorFlow GPU版本上运行): python...–datasets_split_name=train 指定训练数据集 –dataset_dir=satellite/data 训练数据集保存的位置 –model_name=inception_v3 使用的模型名称...inf_graph.pb 文件,但是这个文件不包含训练获得的模型参数,需要将cheeckpoint中的模型参数保存进来,方法是使用freeze_graph.py: python freeze_graph.py

    74410

    使用Tensorflow构建属于自己的图片分类器

    还有我们能够想到的领域,机器翻译、无人驾驶、搜索引擎、美颜…好像都被别人早早布局,现在进入是不是太晚了?...数据生成 首先使用手机拍一段视频,导入到电脑,然后通过软件保存图像。在ubuntu上可以使用ffmpeg软件来完成。.../tensorflow-for-poets-2cd tensorflow-for-poets-2 训练数据集 将前面通过视频生成的图片集放到tf_files目录下,每一类图片单独建一个文件夹文件夹可以如下所示...: tf_files/retrained_graph.pb,再训练的图文件。...至此,训练我们自己的分类器的任务就结束了,在下一篇文章中,我将带领大家探索如何在Android手机上使用我们的图片分类器。

    1.1K60

    tensorflow从ckpt和从.pb文件读取变量的值方式

    最近在学习tensorflow自带的量化工具的相关知识,其中遇到的一个问题是从tensorflow保存好的ckpt文件或者是保存后的.pb文件(这里的pb是把权重和模型保存在一起的pb文件)读取权重,查看量化后的权重是否变成整形...(1) 从保存的ckpt读取变量的值(以读取保存的第一个权重为例) from tensorflow.python import pywrap_tensorflow import tensorflow.../model_ckpt') #保存ckpt文件文件夹 if ckpt and ckpt.model_checkpoint_path: reader = pywrap_tensorflow.NewCheckpointReader....pb文件读取变量的值(以读取保存的第一个权重为例) import tensorflow as tf from tensorflow.python.framework import graph_util...('Yourpb.pb', 'rb') as f: #自己保存pb文件 graph_def = tf.GraphDef() graph_def.ParseFromString(f.read())

    3.6K20

    Tensorflow SavedModel模型的保存与加载

    这两天搜索了不少关于Tensorflow模型保存与加载的资料,发现很多资料都是关于checkpoints模型格式的,而最新的SavedModel模型格式则资料较少,为此总结一下TensorFlow如何保存...variables文件夹保存训练所习得的权重。assets文件夹可以添加可能需要的外部文件,assets.extra是一个库可以添加其特定assets的地方。...saved_model.pb 保存 为了简单起见,我们使用一个非常简单的手写识别代码作为示例,代码如下: from tensorflow.examples.tutorials.mnist import...要保存该模型,我们还需要对代码作一点小小的改动。 添加命名 在输入和输出Ops中添加名称,这样我们在加载时可以方便的按名称引用操作。...,第三个参数是模型保存文件夹

    5.4K30

    精通 TensorFlow 2.x 计算机视觉:第三、四部分

    将所有图像保存在一个文件夹中。...定义源(.jpg文件)和目标(.xml文件)目录。 选择每个图像并在其周围绘制一个矩形。 定义类名称保存。 如果给定图像中有多个类别,则在每个图像周围绘制一个矩形,并为其分配相关的类别名称。...请按照以下步骤执行此操作: 在继续下一步之前,请将Chapter10_Tensorflow-Training_a_Object_Detector_GoogleColab.ipynb Python 文件保存到...,如何生成 TensorFlow .xml和.jpg文件中的tfRecord文件,如何开始训练过程并在训练过程中监控读数,如何创建 TensorBoard 以观察训练准确率,如何在训练后保存模型以及如何通过保存的模型进行推断...在这种情况下,sample1是 S3 存储桶名称,DEMO是其中包含sample1的文件夹,其中有六个文件夹-两个数据文件夹,其中包括.jpg图像,两个标注文件夹由.json文件,输出和检查点文件组成。

    5.7K20

    【技术分享】深入了解tensorflow模型存储格式

    导语 做模型的同学基本都会使用tensorflow,不知道大家是否会像我一样对tensorflow的模型存储感到疑惑:各种模型保存的方法、保存出的模型文件名称和结构还不一样、加载模型的时候有的需要重新定义一遍计算图而有的不需要...这篇文章会带大家了解每个模型文件分别包含什么内容、计算图是以什么样的形式保存文件中的。 以下讨论的api都是基于tensorflow1.15版本。...tensorflow还支持其他函数来进行模型保存采用Estimator的export_savedmodel方法,但保存下来的模型格式总是包含在上面两种格式之内。...3 保存计算图的文件 文章第一章的两种保存格式区别就在于model.meta和saved_model.pb。这两者都保存了模型的计算图,而且这两者都是protobuf文件。...基本的节点能组成复杂的操作,很多tensorflow在python代码中的函数,tf.nn.moments,其实是由很多加法、乘法、求平均等op节点构成的,在pb中并不会出现一个op是moments

    3K30

    tensorflow实现将ckpt转pb文件的方法

    本博客实现将自己训练保存的ckpt模型转换为pb文件,该方法适用于任何ckpt模型,当然你需要确定ckpt模型输入/输出的节点名称。...使用 tf.train.saver()保存模型时会产生多个文件,会把计算图的结构和图上参数取值分成了不同的文件存储。这种方法是在TensorFlow中是最常用的保存方式。...model.ckpt.meta文件保存TensorFlow计算图的结构,可以理解为神经网络的网络结构,该文件可以被 tf.train.import_meta_graph 加载到当前默认的图来使用。...PS:注意节点名称,应包含name_scope 和 variable_scope命名空间,并用“/”隔开,”InceptionV3/Logits/SpatialSqueeze” 二、 pb模型预测 下面是预测...PS:注意张量的名称,即为:节点名称+“:”+“id号”,”InceptionV3/Logits/SpatialSqueeze:0″ 完整的CKPT 转换成 PB格式和预测的代码如下: # -*-coding

    2.5K30

    keras模型保存tensorflow的二进制模型方式

    如何将keras训练好的模型转换成tensorflow的.pb文件并在TensorFlow serving环境调用 首先keras训练好的模型通过自带的model.save()保存下来是 .model...(.h5) 格式的文件 模型载入是通过 my_model = keras . models . load_model( filepath ) 要将该模型转换为.pb 格式的TensorFlow 模型,...格式的文件 问题就来了,这样存下来的.pb格式的文件是frozen model 如果通过TensorFlow serving 启用模型的话,会报错: E tensorflow_serving/core...saved model 文件夹下就有两个文件: saved_model.pb variables 其中variables 可以为空 于是将.pb 模型导入serving再读取,成功!...以上这篇keras模型保存tensorflow的二进制模型方式就是小编分享给大家的全部内容了,希望能给大家一个参考。

    1.1K30

    模型保存,加载和使用

    0x01 TensorFlow模型 1.1 模型文件 TensorFlow模型会保存在checkpoint相关文件中。...因为TensorFlow会将计算图的结构和图上参数取值分开保存,所以保存后在相关文件夹中会出现3个文件。 下面就是DIN,DIEN相关生成的文件,可以通过名称来判别。...: checkpoint文件保存了一个目录下所有的模型文件列表,这个文件TensorFlow自动生成且自动维护的。...当某个保存TensorFlow模型文件被删除时,这个模型所对应的文件名也会从checkpoint文件中删除。...1.2 freeze_graph 正如前文所述,tensorflow在训练过程中,通常不会将权重数据保存的格式文件里,反而是分开保存在一个叫checkpoint的检查点文件里,当初始化时,再通过模型文件里的变量

    1.4K10

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

    升级到tf 2.0后, 训练的模型想转成1.x版本的.pb模型, 但之前提供的通过ckpt转pb模型的方法都不可用(因为保存的ckpt不再有.meta)文件, 尝试了好久, 终于找到了一个方法可以迂回转到...下保存的模型是.h5格式的,并且仅保存了weights, 即通过model.save_weights保存的模型....在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模型...以上这篇tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例就是小编分享给大家的全部内容了,希望能给大家一个参考。

    2.5K20

    可能是最好玩的深度学习模型:CycleGAN的原理与实验详解

    今天这篇文章主要分成三个部分: CycleGAN的原理解析 CycleGAN与原始的GAN、DCGAN、pix2pix模型的对比 如何在TensorFlow中用CycleGAN训练模型 CycleGAN...的原理 我们之前已经说过,CycleGAN的原理可以概述为:将一类图片转换成另一类图片。...在TensorFlow中实验CycleGAN 最后来讲一讲如何在TensorFlow中实验CycleGAN,打开全球最大的同性交友网站Github,我们可以发现CycleGAN在TensorFlow中已经有很多轮子了...在实践中,我使用了CelebA数据集,分别取出其中男性和女性的图片并统一缩放到256x256的大小,然后存入两个文件夹中: ?...woman.tfrecords \ --image_size 256 训练的过程比较漫长,此时可以打开TensorBoard来观察训练情况(运行这个命令时需要将“20170715-1622”改成机器中对应的文件夹

    3.3K40

    如何将自己开发的模型转换为TensorFlow Lite可用模型

    我们来分析一下从训练文件保存的不同的TF格式。...重要的是要注意GraphDef、Saved Model、FrozenGraph和Optimized Graphs都以protobuf格式保存(.pb) 训练模型 - 这将生成3个代表网络结构的文件。...我们关心的是GraphDef和检查点文件。在训练脚本的命令中,保存这些文件文件夹位于/tmp/mnist_saved_model下。...同样,input_tensor之前的所有内容也是不必要的。在移动设备上运行之前,我们需要裁剪此图。 TFLite中大多数训练层也不受支持(请参阅附录)。...附录 使用Tensorboard 我创建了一个修改版本的import_pb_to_tensorboard.py,以支持导入图形定义(.pb)、图形定义文本(.pbtxt)和保存模型(.pb文件类型。

    3K41

    tensorflow:AToolDeveloperGuideToTFModelFIles

    水平有限,发现错误,请不吝指出! 翻译开始 大多数用户不需要关心tensorflow在硬盘上存储数据的细节问题的,但是如果你是一个 Tool developer, 那就另当别论了。...概括来说就是,你在文本文件(text files)中定义数据结构,protobuf tools就会生成对应的C,Python和其它语言的类。我们可以用友好的方式来加载,保存访问这些类中的数据。...Nodes 一旦将文件加载到graph_def对象,你就可以访问内部的数据了。出于实用目的,最重要的部分是存储节点成员的节点列表。...当Graph运行起来的时候,就会在注册表中查找这些op的名称以找到其对应的实现。...='r',如果之前保存的是二进制文件 mode='rb' with open(".

    1.4K50
    领券