如何将一列图像url转换成数据帧中的numpy数组？

Question

我正在Kaggle上处理Plant Seedlings数据集，并准备了一个包含2列的dataframe。在

第一列是列车组中每个图像的目录，第二列是该图像的标签（名称）。在

我想把它转换成一个数据帧，这样我就可以用这个数据帧来训练我的模型了。在

此外，图像有3个通道。在

给定目录和标签为arr的数据帧的名称

                              file               category
0        ../input/train/Maize/a5c2eec2d.png        Maize
1        ../input/train/Maize/8cd93b279.png        Maize
2        ../input/train/Maize/8c6fba454.png        Maize
3        ../input/train/Maize/abadd72ab.png        Maize
4        ../input/train/Maize/f60369038.png        Maize

我应该如何完成上述任务？在

Answer 1

from PIL import Image
import numpy as np

dataset = []
# If you to encode category names you can do the following
# df['category_code'] = df['category'].cat.codes 
# and you can iterate over this in for loop
for image_name, category in zip(df['file'],df['category']):
    image = np.asarray(Image.open(image_name))
    dataset.append((image,category))

将图像调整到特定大小

^{pr2}$

其中size是一个元组（224224224）