繁体   English   中英

Tensorflow:图像预处理盗版v4不接受我的jpg

[英]Tensorflow: Image Preprocessing Inception v4 not accepting my jpg

我是Tensorflow的新手,我尝试使用Tensorflow中提到的脚本:使用预训练的初始模型来避免使用TF记录,但是我所有的预测最终都归于同一错误的类中。 但是,评估分类器会产生正确的结果,因此它不是模型,我认为预处理是我做错的。

因此,我决定尝试使用初始预处理功能,但现在它不接受我的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))
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)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM