使用Inception标签图像获取ValueError：GraphDef不能大于2GB

我使用TensorFlow作为Poets代码实验室来指导我，因为我正在重新培训Inceptionv3 CNN以对图像列表进行分类。我已经成功地训练了模型，并且在我使用给定的代码对单个图像进行分类时，它可以工作。但是当我尝试在大量图像上使用它时，我得到的GraphDef不能大于2GB。请指教。使用Inception标签图像获取ValueError：GraphDef不能大于2GB

import pandas as pd 
import os, sys 
import tensorflow as tf 
test_images = pd.read_csv('test_images.csv') 
testid = test_images['Id'] 
listx= list(range(4320)) 
predlist=[] 
output = pd.DataFrame({'Id': listx}) 
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' 
for x in listx: 
    path = 'test/'+str(x+1)+'.jpg' 

# change this as you see fit 
    image_path = path 

# Read in the image_data 
    image_data = tf.gfile.FastGFile(image_path, 'rb').read() 

# Loads label file, strips off carriage return 
    label_lines = [line.rstrip() for line 
       in tf.gfile.GFile("retrained_labels.txt")] 

# Unpersists graph from file 
with tf.gfile.FastGFile("retrained_graph.pb", 'rb') as f: 
    graph_def = tf.GraphDef() 
    graph_def.ParseFromString(f.read()) 
    tf.import_graph_def(graph_def, name='') 

with tf.Session() as sess: 
    # Feed the image_data as input to the graph and get first prediction 
    with tf.Graph().as_default(): 
     softmax_tensor = sess.graph.get_tensor_by_name('final_result:0') 

     predictions = sess.run(softmax_tensor, \ 
          {'DecodeJpeg/contents:0': image_data}) 

    # Sort to show labels of first prediction in order of confidence 
    top_k = predictions[0].argsort()[-len(predictions[0]):][::-1] 
    # print('the top result is' + label_lines[node_id]) 
    flag = 0 
    for node_id in top_k: 

     while flag == 0: 
      human_string = label_lines[node_id] 
      score = predictions[0][node_id] 
      predlist.append(int(human_string[:3])) 
      print('%s' % (human_string)) 

      flag = 1 # we only want the top prediction

输出[ '预测'] = predlist output.to_csv（ 'outputtest.csv'）由该错误可以通过电子邮件解决

来源

2017-06-02 user3580196

的一种方式是通过将

with tf.Graph().as_default():

后for循环。这是我尝试读取批量图像时为我工作的一段代码：

for filename in os.listdir(image_path): 

     with tf.Graph().as_default(): 
     # Read in the image_data 
     image_data = tf.gfile.FastGFile(image_path + filename, 'rb').read()

来源

2017-06-23 11:24:43 danish

使用Inception标签图像获取ValueError：GraphDef不能大于2GB

回答

相关问题