Keras：如何使用 flow_from_directory 提供自定義標簽？

Question

我正在嘗試使用 keras 創建一個多標簽分類模型。 因此，我將所有圖像都放在一個文件夾中。 此外，我有一個 CSV 文件，其中包含每個圖像的路徑，后跟多個可能的標簽

Example of my CSV:

path, x1, x2, x3    
img/img_00000001.jpg,1,0,1
img/img_00000002.jpg,0,0,1
...

我正在嘗試使用 flow_from_directory 讀取我的圖像並通過 CSV 提供相應的標簽。 我到目前為止看起來像這樣：

image_path= "C:/user/Images"


data_generator = ImageDataGenerator(rescale=1./255, 
                                    validation_split=0.20)

train_generator = data_generator.flow_from_directory(image_path, target_size=(IMAGE_HEIGHT, IMAGE_SIZE), shuffle=True, seed=13,
                                                     class_mode='binary', batch_size=BATCH_SIZE, subset="training")

validation_generator = data_generator.flow_from_directory(image_path, target_size=(IMAGE_HEIGHT, IMAGE_SIZE), shuffle=False, seed=13,
                                                     class_mode='binary', batch_size=BATCH_SIZE, subset="validation")

這里建議了類似問題的解決方案： 如何在 keras flow_from_directory 中手動指定類標簽？ 提供此代碼：

def multiclass_flow_from_directory(flow_from_directory_gen, multiclasses_getter):
    for x, y in flow_from_directory_gen:
        yield x, multiclasses_getter(x, y)

但是，我無法弄清楚如何實現 multiclasses_getter() 使其工作。

Answer 1

嘗試使用 flow_from_dataframe 代替 flow_from_directory