2017-08-06 96 views
0

我是新来tensorflow,我试图用Tensorflow: use pretrained inception model提到的脚本,以避免使用TF纪录,但我所有的预言在同一个错误的类中结束。然而评估分类器会产生正确的结果,所以它不是模型,我相信预处理就是我做错了。Tensorflow:图像预处理启V4不接受我的JPG

所以我决定尝试成立以来预处理功能,但现在它不会接受我的JPG文件。我得到这个错误:

inception_preprocessing.py", line 265, in preprocess_for_eval 
    image = tf.image.central_crop(image, central_fraction=central_fraction) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/image_ops_impl.py", line 335, in central_crop 
    _Check3DImage(image, require_static=False) 
    File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/ops/image_ops_impl.py", line 129, in _Check3DImage 
    raise ValueError("'image' must be three-dimensional.") 
ValueError: 'image' must be three-dimensional. 

这里是我的代码:

arg_scope = inception_utils.inception_arg_scope() 
im_size = 299 

inputs = tf.placeholder(tf.float32, (None, im_size, im_size, 3)) 
inputs = inception_preprocessing.preprocess_image(inputs, im_size, im_size) 

with slim.arg_scope(arg_scope): 
    logits, end_points = inception_v4.inception_v4(inputs) 

saver = tf.train.Saver() 
saver.restore(sess,ckpt_file) 

for image in sample_img: 
    im = Image.open(image) 
    im = im.resize((im_size,im_size)) 
    im = np.array(im) 

    logit_values = sess.run(logits, feed_dict={inputs:im}) 

    print(np.argmax(logit_values)) 
+0

你检查IM的形状? –

回答

0
inputs_processed = inception_preprocessing.preprocess_image(tf.squeeze(inputs), im_size, im_size) 
inputs_processed = tf.expand_dims(inputs_processed, 0) 
with slim.arg_scope(arg_scope): 
    logits, end_points = inception_v4.inception_v4(inputs_processed) 
# placeholder feed value should be 4D 
im = np.exapnd_dims(np.array(im), 0)