如何修复加载函数中的IOError？

Question

我使用flow_from_dataframe函数来扩充我的数据集。在

但是什么时候

train= data_gen.flow_from_dataframe(dataframe=train_df,
                                               directory=None,
                                               x_col='image_name',
                                               y_col='label',
                                               class_mode="raw",
                                               target_size=(img_width, img_height),
                                               batch_size=2048)
for i in train:
    break

为了保存我发现的IOError图像，我对输出进行了迭代。

for i in train:

^{2}$

Answer 1

最后，我解决了这个问题。当我们使用flow_from_dataframe时，我们应该指定颜色模式（默认为RGB）。如果我们的图像处于灰色模式并且我们使用默认设置，则会引发此错误。在