如何將 CK+ 數據集放入 dataframe 中，其中一列是圖像，第二列是情感（這甚至有必要嗎？）

Question

我想構建一個 model 用於情感分類，並且我正在努力處理數據集。 我正在使用 CK+，因為我讀到它符合行業標准。 我不知道如何以正確的方式格式化它，所以我可以開始工作。 數據集按以下方式格式化。

憤怒（文件夾）

• 文件 1
• 文件 2
• ...

蔑視（文件夾）

• 文件 3
• 文件 4
• ...

我需要文件夾名稱作為文件夾內文件的標簽，但我真的不知道如何到達那里。

Answer 1

您可以使用tf.keras.utils.image_dataset_from_directory function 將所有數據加載到tf.data.Dataset中。 假設您的Anger和Contempt文件夾位於名為Parent的目錄中，您可以這樣做：

import tensorflow as tf
dataset = tf.keras.utils.image_dataset_from_directory('Parent')

然后，您可以直接從數據集中訪問圖像和標簽，例如：

iterator = dataset.as_numpy_iterator()
print(iterator.next())