使用 Galaxy Zoo 数据集、TensorFlow 和 Keras 训练 GAN

Question

为了使用 TensorFlow 和 Keras 训练简单的 GAN，导入 Galaxy Zoo dataset 的正确方法是什么？例如，是否有与导入 MNIST 相同的简单方法？

from keras.datasets import mnist

Answer 1

没有简单的导入，因为它不是 keras dataset。

您必须自己下载数据并进行预处理，然后再将其送入网络。如果你去 kaggle link 和 notebooks，你可以看到其他人如何预处理他们的数据进行训练

重要的是你想训练什么以及你在做什么class。如果您想在图像上进行训练以 class 化 csv 中的全部或部分 class ，您将需要将数据作为图像并 class 化作为标签。如果您想使用一些 class 化来预测另一个 class 化，一些 class 将是数据而其他人将是您预测的标签。

我建议将训练 csv 导入 pandas dataframe

import pandas as pd
df = pd.read_csv('../input/44352/training_solutions_rev1.csv')

然后您必须为您的图像创建一个自定义生成器，因为它们在文件夹结构中不是按 class 组织的。