将图像添加到熊猫数据帧行

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)这将以迭代方式向您创建的空数据帧添加一行，您现在所做的只是在每次迭代时使用单个图像创建一个新数据框