-
大小: 2KB文件类型: .py金币: 1下载: 0 次发布日期: 2021-05-06
- 语言: Python
- 标签: tensorflow pb
资源简介
使用这个python程序可以直接将ckpt文件转成固化的模型文件,用于预测结果,注意修改路径
代码片段和文件信息
import tensorflow as tf
#from create_tf_record import *
from tensorflow.python.framework import graph_util
def freeze_graph(input_checkpointoutput_graph):
‘‘‘
:param input_checkpoint:
:param output_graph: PB模型保存路径
:return:
‘‘‘
# checkpoint = tf.train.get_checkpoint_state(model_folder) #检查目录下ckpt文件状态是否可用
# input_checkpoint = checkpoint.model_checkpoint_path #得ckpt文件路径
# 指定输出的节点名称该节点名称必须是原模型中存在的节点
output_node_names = “InceptionV1/Logits/Predictions/Reshape_1“
saver = tf.train.import_meta_graph(input_checkpoint + ‘.meta‘ clear_devices=True)
graph = tf.get_default_graph() # 获得默认的图
input_graph_def = graph.as_graph_def() # 返回一个序
相关资源
- tensorflow-C3D-ucf101网络
- lstm_tensorflow
- 《TensorFlow2深度学习》
- EasyPBC V.1.3 easy PBC插件
- TensorFlow Python API documentation.pdf
- Python-使用最新版本的tensorflow实现se
- BP神经网络及代码分析(python+tensorf
- Python-用TensorFlow实现神经网络实体关系
- 莫烦全部代码Reinforcement-learning-with-
- Python-基于TensorFlow和BERT的管道式实体
- Tensorflow gpu_accelerate
- python tensorFlow AND和XOR
- word2vec.py(来自黄文坚的“tensorflow实
- keras+tensorflow CNN
- 基于机器学习框架tensorflow的图像分类
- 机器学习实战:基于 Scikit-Learn 和 T
- 《白话深度学习与TensorFlow》.pdf
- Tensorflow+实战Google深度学习框架
- python MNIST分类 tensorflow
- 树莓派Raspberry Pi实战指南-手把手教你
- tensorflow下用LSTM网络进行时间序列预测
- 深度学习之二:用Tensorflow实现卷积神
- tensorflow版本的YOLO v3,在Windows系统下
- Python实现的TensorFlow入门案例
- 《TensorFlow技术解析与实战》高清中文
- python实现TensorFlow2 股票股价预测(源
- 计算 语义分割结果 MIOU,miou.py操作简
- tensorflow做cifar-10识别
- TensorFlow手写数字识别代码
- 基于kNN方法的MNIST手写数字识别Tenso
评论
共有 条评论