Как преобразовать столбец URL-адресов изображений в массивы в массиве данных?

Question

Я работаю над набором данных Plant Seedlings в Kaggle, и я подготовил фрейм данных, содержащий 2 столбца.

В первом столбце есть каталог каждого изображения, присутствующего в наборе поездов, и второй столбец.имеет метку (имя) этого изображения.

Я хочу преобразовать его в фрейм данных таким образом, чтобы затем использовать этот фрейм данных для обучения моей модели.

Кроме того,В изображении есть 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))

Для изменения размера изображения до определенного размера,

image = np.asarray(Image.open(image_name).resize(size))

, где размер соответствует кортежу (224,224)