tensorflow - 使用 flow_from_dataframe y_col 的正确 "value"是什么

Question

我正在用 Pandas 读取 csv 文件，并给出存储在 colname 中的列名

colnames=['file', 'label'] 
# Read data from file 
data = pd.read_csv('./Hand_Annotations_2.csv',names=colnames, header=None)
# Preview the first 5 lines of the loaded data 
data.head()

然后，我使用 ImageDataGenerator()和 flow_fromdataframe()获取批量数据

train_generator=datagen.flow_from_dataframe(dataframe=data,
                                            directory=None, 
                                            x_col=colnames[0], 
                                            y_col=colnames[1],
                                            class_indices=IDmap,
                                            class_mode="categorical", target_size=(224,224), batch_size=32)

但我收到一个错误，如下所示:

TypeError: If class_mode="categorical", y_col="label" column values must be type string, list or tuple.

但是我的 y_col是一个字符串。如果我只输入 "label"，我会得到同样的错误.它似乎也适用于 x_col .

有人可以指出我的错误吗？

谢谢

解决方案

使用 dtype 读取 csv，例如字符串:

data = pd.read_csv('./Hand_Annotations_2.csv',dtype=str,names=colnames, header=None)

Answer 1

你可以检查github中的代码以了解实际检查的内容，特别是here . Keras 预处理正在检查所有值是否具有允许的类型(str、list 或 tuple)，因此检查失败，您必须检查 y_col 的所有值列有正确的类型。