调整图像大小以在TensorFlow中进行训练

Question

我尝试在 TensorFlow 中读取自己的图像进行训练，但是似乎出现了错误:

ValueError: 'size' must be a 1-D Tensor of 2 elements: new_height, new_width. 

以下代码有什么问题？

filenames=['images/000001.jpg','images/000002.jpg','images/000003.jpg','images/000004.jpg']
labels=[1,0,1,0]

filename_queue=tf.train.string_input_producer(filenames)

reader=tf.WholeFileReader()
filename, content = reader.read(filename_queue)
images=tf.image.decode_jpeg(content, channels=3)
images=tf.cast(images, tf.float32)
resized_images=tf.image.resize_images(images, 224, 224)

image_batch, label_batch=tf.train.batch([resized_images, labels], batch_size=2)

- cerebrou

---

什么是错误？请提供它。 - Mox

Answer 1

这个错误提示的意思是，size 必须是一个一维张量。TensorFlow 实际上是希望你将 tf.image.resize_images 的第二个参数改为一个元组：

resized_images = tf.image.resize_images(images, (224, 224))

- cerebrou

Answer 2

我遇到了同样的错误，将元组转换为TensorFlow常量即可解决

resized_images = tf.image.resize(images, tf.constant([224, 224]))

- Yusuf Syam