Question

我训练了一个CNN模型来对图像中的两个对象进行分类。训练结束后，我得到了thos文件：模型80.data 00000-的-00001 模型80.index 模型80.meta

我想创建预测脚本因此我看到我需要将图形冻结为.pb文件，但我遇到了这个问题：

将0变量转换为const ops。

我的冻结代码是：

HttpSession session = request.getSession();
 String ip_address = session.getAttribute("ip_address");

以：

运行

import time
import math
import random
import os
from tensorflow.python.framework import graph_util

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
import dataset
import cv2

saver = tf.train.import_meta_graph('model/model-80.meta', clear_devices=True)
graph = tf.get_default_graph()
input_graph_def = graph.as_graph_def()
sess = tf.Session()
saver.restore(sess, "model/model-80")

output_node_names="x,y_true"
output_graph_def = graph_util.convert_variables_to_constants(
            sess, # The session
            input_graph_def, # input_graph_def is useful for retrieving the nodes 
            output_node_names.split(",")  
)

output_graph="model80.pb"
with tf.gfile.GFile(output_graph, "wb") as f:
    f.write(output_graph_def.SerializeToString())

sess.close()

我的cnn代码是：

 python export.py --checkpoint_path=model --output_nodes=x,y_true

由于

将CNN图模型冻结到.pb文件

0 个答案: