当使用来自数据帧的flow_读取数据时,Keras会安装ImageDataGenerator

2024-04-25 11:41:16 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试使用ImageDatGenerator的以下选项

data_generator = ImageDataGenerator(featurewise_center=True, featurewise_std_normalization=True)

这将需要在使用它之前运行以下行

data_generator.fit(train_data)

然而,我的训练数据信息是在一个数据帧中,我用flow_from_dataframe函数读取它,如下所示

data_generator.flow_from_dataframe(dataframe=data, directory=imgs_path, x_col="filename", y_col="class",
                                              class_mode="binary", target_size=(img_width,img_height),
                                              batch_size=batch_size)

因此,在调用flow_from_dataframe函数之前,我没有适合生成器的训练数据。有什么解决办法吗