python - 使用 python 从目录加载图像并 reshape

Question

<分区>

Answer 1

假设您有 scipy安装并假设“reshape”实际上是指“resize”，下面的代码应该加载目录 /foo/bar 中的所有图像，将它们调整为 64x64 并将它们添加到列表 图片:

import os
from scipy import ndimage, misc

images = []
for root, dirnames, filenames in os.walk("/foo/bar"):
    for filename in filenames:
        if re.search("\.(jpg|jpeg|png|bmp|tiff)$", filename):
            filepath = os.path.join(root, filename)
            image = ndimage.imread(filepath, mode="RGB")
            image_resized = misc.imresize(image, (64, 64))
            images.append(image_resized)

如果您需要一个 numpy 数组(调用 reshape)，那么只需在末尾添加 images = np.array(images)(使用 import numpy as np 开头)。