试图将两个 MapDataset 组合成一个 MapDataset

Question

我正在尝试将这两个地图数据集合并为一个。 换句话说，我正在尝试扩展我的数据集。

ds = ds.map(lambda x, y: (load_aug(tf.image.resize(x, size)), y))
ds1 = ds1.map(lambda x, y: (tf.image.resize(x, size), y))

ds 是数据集的增强版本，ds1 是常规的未增强数据集。 两个数据集中的图像形状相同。

Answer 1

您可以使用zip或concatenate来组合两个 MapDataset：

import tensorflow as tf

ds = tf.data.Dataset.range(1, 4)
ds1 = tf.data.Dataset.range(4, 8) 

train_dataset = tf.data.Dataset.zip((ds, ds1))
# or 
train_dataset = ds.concatenate(ds1)