tf生成批数据集

ds_train = tf.data.Dataset.list_files("./data/cifar2/train/*/*.jpg") \
.map(load_image) \
.shuffle(buffer_size=2000)\
.batch(100) \
.prefetch(-1).repeat(3)