今天早上我一直想让TensorFlow io api工作.
经过一些研究,我设法读取了数据,但是我无法在出列时正确绑定图像和标签.
这是我写的代码:
# load csv content
csv_path = tf.train.string_input_producer(['list1.csv', 'list2.csv'])
textReader = tf.TextLineReader()
_, csv_content = textReader.read(csv_path)
im_name, label = tf.decode_csv(csv_content, record_defaults=[[""], [1]])
# load images
im_content = tf.read_file(im_dir+im_name)
image = tf.image.decode_png(im_content, channels=3)
image = tf.cast(image, tf.float32) / 255.
image = tf.image.resize_images(image, 640, 640)
# make batches
im_batch, lb_batch = tf.train.batch([image, label], batch_size=batch)
Run Code Online (Sandbox Code Playgroud)
的顺序im_batch和lb_batch被弄乱了(将图像结合到随机标签).
知道发生了什么事吗?谢谢.