У меня проблема с хранением файлов изображений в массиве для дальнейшего обучения модели классификации.
Проблема: у меня есть кадр данных со столбцом ссылок (назовем его photos_link), мне нужно прочитать ссылку, получить изображение (все размеры одинаковы), преобразовать его в массив, добавить в список, затем преобразовать этот список в np.array. в результате этого скрипта у меня должен быть массив с формой: (% количество фотографий%,% высота фото%,% ширина фото%,% 3-, который показывает, что это RGB-изображение%). Но я получаю только эту форму: (% количество фотографий%,), пожалуйста, скажите мне, где я не прав. Код ниже:
links = [link1,link2,link3,link4,link5,link6,link7,link8,link9,link10]
train_images = []
for each in links:
img = io.imread(each)
img = rescale(img, 0.25, anti_aliasing=False)
img = image.img_to_array(img)
train_images.append(img)
X_test = np.array(train_images)
X_test.shape output: (10,)
Заранее спасибо