我使用TensorFlow函数tf.image.resize_images来调整图像大小,但是我在代码中遇到了这个错误: ValueError:'images'不包含任何形状。 完整代码如下:
# -*- coding: utf-8 -*-
import tensorflow as tf
file = ["./1.jpg"]
f = tf.train.string_input_producer(file)
reader = tf.WholeFileReader()
key, img = reader.read(f)
img = tf.image.decode_image(img)
# img.set_shape([218,178,3])
img = tf.image.resize_images(img, [64,64])
coord = tf.train.Coordinator()
with tf.Session() as sess:
tf.train.start_queue_runners(coord=coord)
image = sess.run(img)
完整的错误信息是
Traceback (most recent call last):
File "image_read_test.py", line 10, in <module>
img = tf.image.resize_images(img, [64,64])
File "C:\Python35\lib\site-packages\tensorflow\python\ops\image_ops_impl.py", line 724, in resize_images
raise ValueError('\'images\' contains no shape.')
ValueError: 'images' contains no shape.
然后我尝试修复此问题,但只找到类似的方式
# -*- coding: utf-8 -*-
import tensorflow as tf
file = ["./1.jpg"]
f = tf.train.string_input_producer(file)
reader = tf.WholeFileReader()
key, img = reader.read(f)
img = tf.image.decode_image(img)
# img.set_shape([218,178])
# img = tf.image.resize_images(img, [64,64])
coord = tf.train.Coordinator()
with tf.Session() as sess:
tf.train.start_queue_runners(coord=coord)
image = sess.run(img)
image = tf.image.resize_images(image, [64,64])
只有这样功能才能正常运行,但我不知道为什么?函数tf.image.resize_images是否只将numpy数组作为参数?或者我可以找到解决这个问题的另一种方法?注意:img.set_shape([218,78,3])对我不起作用
答案 0 :(得分:15)
传递expand_animations = False
作为参数很重要:
尝试:
tf.image.decode_image(img, expand_animations = False)
确保具有3维形状的张量。 此问题归因于gif格式,因为与其他格式返回3D数组[height,width,num_channels]的其他格式(包括decode_bmp,decode_jpeg和decode_png)相反,decode_gif返回4D数组[num_frames,height,width,3]。
有关更多信息,请检查related documentation
答案 1 :(得分:10)
我最近遇到了这个问题,
tf.image.decode_image()
没有返回形状的张量,但我的图像都是jpeg格式。
所以我用了
tf.image.decode_jpeg()
它返回形状的张量,解决了问题。 请注意, tf.image.decode_png()也是如此。
答案 2 :(得分:3)
image_string = tf.read_file(filename)
image_decoded = tf.cond(
tf.image.is_jpeg(image_string),
lambda: tf.image.decode_jpeg(image_string, channels=3),
lambda: tf.image.decode_png(image_string, channels=3))