试图将两个 MapDatasets 组合成一个 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 组合两个 MapDatasets：

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)