將圖像添加到 Pandas 數據框行

Question

我想從文件夾加載圖像，將它們轉換為 numpy 數組，然后將它們存儲為數據幀行。 我應該如何進行？

這是我想出的：

from keras.preprocessing.image import load_img
from keras.preprocessing.image import img_to_array
from keras.preprocessing.image import array_to_img

BASE_DIR = '/kaggle/input/coughspectrograms/content/'
os.chdir('/kaggle/input/coughspectrograms/content/')

files_in_train = sorted(os.listdir(BASE_DIR))
files_in_train.pop(0)
img_df = pd.DataFrame()

for img in files_in_train: 
    loaded_img = load_img(img)
    img_array = img_to_array(loaded_img)

    img_df = pd.DataFrame({"image": [img_array]})

Answer 1

如果您想要最終的數據框形狀為 [n_images, 1]，您只需將代碼中的最后一行替換為：

img_df = img_df.append({"image": [img_array]}, ignore_index=True)

這將迭代地向您創建的空數據框添加一行，您現在所做的只是在每次迭代時創建一個帶有單個圖像的新數據框