这里是代码:
import tensorflow as tf
slim = tf.contrib.slim
import PIL as pillow
from PIL import Image
#import Image
from inception_resnet_v2 import *
import numpy as np
with open('imagenet1000_clsid_to_human.txt','r') as inf:
imagenet_classes = eval(inf.read())
def get_human_readable(id):
id = id - 1
label = imagenet_classes[id]
return label
checkpoint_file = './inception_resnet_v2_2016_08_30.ckpt'
#Load the model
sess = tf.Session()
arg_scope = inception_resnet_v2_arg_scope()
input_tensor = tf.placeholder(tf.float32, [None, 299, 299, 3])
with slim.arg_scope(arg_scope):
logits, end_points = inception_resnet_v2(input_tensor, is_training=False)
saver = tf.train.Saver()
saver.restore(sess, checkpoint_file)
def classify_image(sample_images):
classifications = []
for image in sample_images:
im = Image.open(image).resize((299,299))
im = np.array(im)
im = im.reshape(-1,299,299,3)
im = 2*(im/255.0)-1.0
predict_values, logit_values = sess.run([end_points['Predictions'], logits], feed_dict={input_tensor: im})
#print (np.max(predict_values), np.max(logit_values))
#print (np.argmax(predict_values), np.argmax(logit_values))
label = get_human_readable(np.argmax(predict_values))
predict_value = np.max(predict_values)
classifications.append({"label":label, "predict_value":predict_value})
return classifications
运行某些图像时出现以下错误:
"ValueError: can not reshape array of size 357604 into shape (299,299.3)"
我不知道它的来源。实际上,图像在重塑之前已被调整大小(299、299)。我不明白,因为我的大多数图像都能正常工作,除了少数...
您对导致此问题的原因有任何想法吗?
提前谢谢您:)
SOLUTION:
我使用此处提供的代码将图像转换为RGB:Convert RGBA PNG to RGB with PIL
现在一切正常:)非常感谢您的帮助!
您可以使用PIL转换轻松地将RGBA((299,299,4)形状)转换为RGB(299,299,3)
im = im.convert('RGB')