python - 多个图像存储在数组中

Question

我面临着将图像文件存储在数组中以进行进一步分类模型训练的问题。

问题：我有一个带有链接列的数据框（我们称之为 photos_link），我需要读取链接，获取图像（所有大小都相同），将其转换为数组，附加到列表，然后将此列表转换为 np 。大批。作为这个脚本的结果，我应该有一个形状为的数组：（照片的%数量%，照片的%高度，照片的%宽度，%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,)

提前致谢

标签： pythonarraysnumpyscikit-image

Answer 1

Looks like X_test is and array of arrays.

X_test[0].shape Will return a shape of (y,x,3) if you in fact have an array of arrays.