У меня есть следующая структура файла CSV ...
image_id, class, sub_class, label1, label2, label3, color
34453572, big, rectangle, 1, 0, 1, red
42524543, small, circle, 1, 1, 0, black
У меня есть следующий код Python:
def read_training_data():
df = pd.read_csv('train.csv', delimiter=',')
labels = ['class', 'sub_class', 'label1', 'label2', 'label3', 'color']
datagen = ImageDataGenerator(rotation_range=30, horizontal_flip=True)
train_generator = datagen.flow_from_dataframe(dataframe=df, directory="train_set", x_col="tag_id", y_col=labels, has_ext=False, class_mode="categorical", target_size=(64, 96), batch_size=32)
return train_generator
Когда параметр y_col = 'что-то' (один классили классификация меток) это работает
, но когда y_col = метки, это дает мне следующую ошибку ошибка
Есть ли способ исправить это?Или есть другой способ подготовки данных ...