我正在使用 pytorch 构建 DCGAN，我的目标是在自定义数据集上进行训练。

我已经发布了一个关于在小型数据集上训练 DCGAN 的问题，当然答案是数据增强。

但我有一个问题。我在 PyTorch 中编码，我想对代表我加载的数据集的现有 ImageFolder 对象执行几种不同的转换。

想法是执行，让我们说大约 5 种不同的转换，并且在执行每个转换之后，我想通过添加新转换的图像来扩展我的数据集。这是伪代码：

list_of_transformations = ['random_crop','center_crop','color_jitter','mirror','grayscale']
dataset = ImageFolder('path_to_dataset')
for t in list_of_transformations
       dataset += dataset.t()

我猜这可能不是最好的主意，甚至被认为是“愚蠢的”，但总的来说我是深度学习和 Python 的新手。任何帮助，将不胜感激！先感谢您！