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

【问题标题】：Trying to combine two MapDatasets to become one MapDataset试图将两个 MapDataset 组合成一个 MapDataset
【发布时间】：2021-12-31 20:48:54
【问题描述】：

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

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 是常规的未增强数据集。两个数据集中的图像形状相同。

【问题讨论】：

问题是什么？
@Kris 我正在尝试将 ds 和 ds1 合并为一个地图数据集

标签： python tensorflow concatenation tensorflow-datasets data-augmentation

【解决方案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)

【讨论】：

非常感谢！顺便说一句，如果我想将它们混合在一起而不是在最后连接呢？
只需与train_dataset = tf.data.Dataset.zip((ds, ds1)) 一起压缩，然后使用train_dataset.map(your_mix_up_function) 做任何你想做的事情。