• 大小: 2KB
    文件类型: .zip
    金币: 1
    下载: 0 次
    发布日期: 2021-01-09
  • 语言: 其他
  • 标签: tensorflow  pb  

资源简介

将tensorflow的PNet,RNet,ONet的模型文件:*.data-00000-of-00001,*.index 和 *.meta这3个文件固化为pb文件。里面详细介绍了节点名称:output_node_names如何获取。直接拿来就可以运行。可以直接看看自己模型文件的所有节点名称。代码里有详细说明

资源截图

代码片段和文件信息

# -*-coding: utf-8 -*-
“““
    @Project: tensorflow_models_nets
    @File   : convert_pb.py
    @Author : panjq
    @E-mail : pan_jinquan@163.com
    @Date   : 2018-08-29 17:46:50
    @info   :
    -通过传入 CKPT 模型的路径得到模型的图和变量数据
    -通过 import_meta_graph 导入模型中的图
    -通过 saver.restore 从模型中恢复图中各个变量的数据
    -通过 graph_util.convert_variables_to_constants 将模型持久化
“““
import numpy as np
import tensorflow as tf
from tensorboard.plugins.beholder.im_util import read_image
from tensorflow.python.framework import graph_util

resize_height = 299  # 指定图片高度
resize_width = 299  # 指定图片宽度
depths = 3


def freeze_graph_test(pb_path image_path):
    ‘‘‘
    :param pb_path:pb文件的路径
    :param image_path:测试图片的路径
    :return:
    ‘‘‘
    with tf.Graph().as_default():
        output_graph

 属性            大小     日期    时间   名称
----------- ---------  ---------- -----  ----
     文件        6796  2019-07-10 15:16  model2pbFile.py

评论

共有 条评论