TensorFlow将ckpt模型转为pb模型

最新推荐文章于 2024-07-08 08:22:55 发布

濯君

最新推荐文章于 2024-07-08 08:22:55 发布

阅读量1.6k

点赞数

分类专栏：机器学习 tensorflow

本文链接：https://blog.csdn.net/zzldm/article/details/104551167

版权

# -*- coding: utf-8 -*-
import tensorflow as tf
from tensorflow.python.framework import graph_util
import os
os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
os.environ['CUDA_VISIBLE_DEVICES'] = '1'
import cv2
import numpy as np


def freeze_graph_test(pb_path, image_path):
    '''
    :param pb_path:pb文件的路径
    :param image_path:测试图片的路径
    :return:
    '''
    with tf.Graph().as_default():
        output_graph_def = tf.GraphDef()
        with open(pb_path, "rb") as f:
            output_graph_def.ParseFromString(f.read())
            tf.import_graph_def(output_graph_def, name="")
        with tf.Session()