如果我的标签也是文件路径，我可以 flow_from_dataframe 吗？

Question

我正在尝试训练一个图像重建网络，我会这样训练：

vae.fit(X_train, X_train, epochs=10, batch_size=1)

其中X_train 是训练图像的 NumPy 数组。

但是，我想使用生成器，否则会耗尽内存。我尝试使用flow_from_dataframe，其中存储了图像的所有文件路径（它们位于多个文件夹中）。

train_generator=datagen.flow_from_dataframe(
                                   dataframe=df,
                                   x_col="filepath",
                                   y_col="filepath")

问题是这个函数输入x_col（文件路径）和y_col（标签）。由于我的损失函数是基于重建错误的，所以我的标签本身应该是相同的图像。有没有办法用这个函数或另一种生成器来做到这一点？

Answer 1

对于自动编码器，您可以设置class_mode=input，然后您不必设置y_col。

所以试试这个：

train_generator=datagen.flow_from_dataframe(
                                   dataframe=df,
                                   x_col="filepath",
                                   class_mode="input")