我尝试读取自己的图像以在 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)
错误说，size必须是一维张量。张量流实际上的意思只是提出第二个论点tf.image.resize_images一个元组：
resized_images = tf.image.resize_images(images, (224, 224))
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
