TensorFlow模型保存的一个坑

最新推荐文章于 2023-02-18 03:26:54 发布

InsZVA

最新推荐文章于 2023-02-18 03:26:54 发布

阅读量2.7k

点赞数

分类专栏：算法文章标签： go python tensorflow 随机森林

本文链接：https://blog.csdn.net/InsZVA/article/details/78616902

版权

最近工作需要，使用tensorflow(r1.3)中contrib的某个库，根据tf官网的说法，go版本是只能调用py版本训练好的模型，而个人对go比较熟且线上环境用go作服务，所以想用go调python训练好的模型。

使用python import tf和tf.contrib相关库，使用tf.saved_model.builder.SavedModelBuilder保存模型，但使用go使用LoadSavedModel的时候，会报形如Not found Op type not registered 'CountExtremelyRandomStats' 的错误，具体的话是说在libtensorflow里面找不到CountExtremelyRandomStats 这个op。

尝试了很多方案，略过。最终发现问题是TensorFlow在BUILD target:libtensorflow.so的时候，根本没有把contrib相关内容打进去。修改根BUILD文件的libtensorflow.so部分如下：

cc_binary(
    name = "libtensorflow.so",
    linkopts = select({
        "//tensorflow:darwin": [
            "-Wl,-exported_symbols_list",  # This

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

InsZVA

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
复制链接

分享到 QQ

分享到新浪微博

扫一扫

专栏目录

TensorFlow 多模型加载融合的诸多坑点

猫ER

11-19

932

TensorFlow在同时加载模型时遇到的问题，以及一些处理办法。

tensorflow 模型保存与加载

最新发布

sdhdsf132452的博客

02-18

871

tensorflow2

解决tensorflow模型保存时Saver报错:TypeError: TF_SessionRun_wrapper: expected all values in input dict to be n...

aids1990的博客

07-20

1521

TypeError: TF_SessionRun_wrapper: expected all values in input dict to be ndarray 对于下面的实际代码: import tensorflow as tf import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' def myregress...

TensorFlow pb模型修改和优化

LuchangLi 的专栏

05-11

2895

TensorFlow 模型训练完成后，通常会通过frozen过程保存一个最终的pb模型。保存的pb模型是以GraphDef数据结构保存的，可以序列化保存为二进制pb模型或者文本pbtxt模型。GraphDef本质上是一个DAG有向无环图，里面主要是存放了一个算子node list，每个算子具有名称，attr等内容，以及通过input包含了node之间的连接关系。整个GraphDef的输入节点是以Placeholder节点来标识的，模型参数权重通常是以Const节点来保存的。不同于onnx，G...

tensorflow 训练保存模型4 自由修改模型

masbbx123的博客

12-19

2995

上回已经可以得到一个清晰的模型结构，但是还是不够满意，为什么呢，因为预测的时候不需要dropout层，所以想修改接口，直接去除dropout层。以下方法自己想的，可能有别的更好的方法：首先，训练的时候得把每层的参数起好名字，否则名字都自动起的，怎么修改网络，直接就懵逼了。 #定义两个placeholder x = tf.placeholder(tf.float32,[None,784]) y ...

tensorflow之模型保存

Richard的专栏

12-02

285

1、ckpt 2、pd def freeze_graph(input_checkpoint,output_graph): ''' :param input_checkpoint: :param output_graph: PB模型保存路径 :return: ''' # 指定输出的节点名称,该节点名称必须是原模型中存在的节点 # 直接用最...

TensorFlow——模型保存与使用

Sam's blog

09-29

621

1、ckpt_save import tensorflow as tf import tensorflow.examples.tutorials.mnist.input_data as input_data mnist = input_data.read_data_sets("MNIST_data/", one_hot=True) x = tf.placeholder(tf.float32, [None, 784], name='input') y_actual = tf.placeholder(t

c语言调用tensorflow模型,<转载>用c++去访问tensorflow模型

weixin_31898831的博客

05-20

884

OpenCV调用TensorFlow预训练模型【尊重原创，转载请注明出处】https://blog.csdn.net/guyuealian/article/details/80570120强大OpenCV从自OpenCV 3.1版以来，dnn模块一直是opencv_contrib库的一部分，在3.3版中，它被提到了主仓库中。新版OpenCV dnn模块目前支持Caffe、TensorFlow、To...

《tensorflow训练好模型移植到android》教程共66页（适合初学者，能很快上手），编译成APK，运行到Android客户端，包含两个实例

01-30

《tensorflow训练好模型移植到android》教程，共66页pdf格式文档（幻灯片形式），讲解清楚，能很快实现。涉及两个例子，第一个是将Tensor flow的实例编译成APK，安装运行到Android客户端，立即可以使用；第二个是自己训练Tensorflow模型，训练几千次之后，将训练好的模型（文档中给出了详细的模型训练，具体训练的图片安装说明再准备提供即可使用），编译成APK，安装运行到Android客户端。适合初学者，能很快上手。具体实验环境为 Mac OS 10.13.2；Tensorflow-master；androidsdk：api_level = 26, build_tools_version = "27.0.3"；androidndk：android-ndk-r12b，api_level=14 . 另外，第一个编译好的APK实例可以在http://download.csdn.net/download/wuxinxing1981/10232528上下载

Tensorflow 保存HDF5模型，并转换为tflite格式

u014798590的博客

12-13

2660

0、引言由于nnc模型量化工具对tflite格式支持较好，同时还支持.caffemodel与.onnx格式（未测），故需在电脑上将训练完成的模型转换为tflite格式后再进行量化，生成kmodel格式模型供K210单片机使用。有关ncc_0.2相关使用，可以见该篇文章（留坑），或详见GitHub网页 1、保存HDF5模型以下代码为HDF5手写数字识别的代码，模型采用全连接层构成，使用relu为激活函数，使用“Dropout”和“提前停止”两种方法防止模型过拟合，同时增加模型泛化能力。需要注意的是，由

Tensorflow 模型保存、节点修改以及Serving 图优化

珍妮的选择的博客

08-14

1615

本文介绍 Tensorflow 模型部分保存方式, 主要包含 `checkpoint` 格式、`frozen_graph` 格式(`SavedModel` 格式暂略), 通过代码实例了解模型的保存方式, Serving 图的优化以及对 Serving 图中的节点进行修改更新.

Tensorflow迁移学习加载预训练模型并修改变量shape和value的方法

lujian1989的专栏

02-16

3638

迁移学习在深度学习中是经常被使用的方法，指的是在一个任务中预训练的模型被用于另一个任务的模型训练，以加快模型训练，减少资源消耗。然而网络搜索相关的话题，基本上只涉及加载预训练模型的特定变量值的方法，即不涉及预训练模型某个变量与当前任务网络对应变量shape改变的处理。在具体的语音合成多说话人模型迁移学习得到单说话人模型的任务中，就涉及到了迁移变量改变shape的情况，将解决方法如下列出。 ...

tensorflow保存模型、加载模型、修改模型、保存修改后的模型、使用修改后的模型做推理、模型转pb、使用pb做推理

broliao的博客

03-31

2610

本文参考于博客，但是增加了自己的一些见解和修改，主要实现的是tensorflow保存模型、加载模型、修改模型、保存修改后的模型和使用修改后的模型做推理。 1.创建和保存模型： import tensorflow as tf w1 = tf.placeholder("float", name="w1") w2 = tf.placeholder("float", name="w2") b1 = tf...

TensorFlow训练模型及保存数量限制的解决办法

DestinyRen的博客

05-07

8262

每次卷积神经网络训练的结果都只保存了最后一部分，查阅了一下相关资料，发现是定义saver时采用的默认值，这里进行如下设置：saver = tf.train.Saver(max_to_keep=100,keep_checkpoint_every_n_hours=1)...

[tf_serving] 在docker中启动tensorflow_serving

currycode

03-11

979

首先在PC的命令行下将model的.pb文件copy到容器中： cd <path to serving model files> sudo docker cp ./serving_am tf_container_asr:/serving 然后进入容器的命令行启动相关服务： root@2effa7569ce3:/serving# tensorflow_model_server...

使用Docker和TensorFlow Serving部署服务

weixin_42530066的博客

08-07

1730

目录 1.启动docker 2.模型由来 3.部署模型 1）拉取镜像 2）启动服务 3）查看模型状态 4）查看模型的输入输出 4.模型接口预测：REST 1.启动docker 安装详细介绍可参考：https://blog.csdn.net/weixin_42530066/article/details/107850996 2.模型由来此处使用已训练好的model，需注意TensorFlowServing是使用SavedModel...

TensorFlow模型保存与加载错误排查

在类`ModelV`和`ModelP`中，`self.saver=tf.train.Saver()`创建了一个Saver对象，用于保存模型的变量。`self.saver.save(self.sess,self.save_path)`这一行代码会将当前Session中的变量保存到指定路径（`self.save_...