tensorflow 利用tf.dataset.Dataset构建数据batch

tf.dataset.Dataset用过多次了，一直懒得记录，今天来更新下

定义阶段

import tensorflow as tf
import numpy as np
...
def image_read(imnames,labs = None):
    
    print(imnames)
    images = tf.read_file('./reptile/combine_total_224/'+imnames)
    images = tf.image.decode_jpeg(images, channels=3)
    images = tf.image.resize_images(images, [224, 224])
    images = tf.cast(images, tf.float32) / 255.0
    images = tf.multiply(images, 2)
    images = tf.subtract(images, 1.0) 
    return images,labs
    
with tf.name_scope('input'):
    image_x = tf.placeholder(imagename_array.dtype,[None],name='image_x')
    label_y = tf.placeholder(label_agen_array.dtype,[None,7,7,6],name='label_y')
    istraing = tf.placeholder(tf.bool,name='istraing')
    
batch_size = 32
dataset = tf.data.Dataset.from_tensor_slices((image_x, label_y)) #对placeholder元组进行切片
dataset &#