python图像到像素的转换性能

Question

我试图在调整大小后将 jpeg 转换为像素。该程序适用于较少的图像。当我尝试使用更多图像（例如 10k）时，它会运行 2 到 3 个小时。有没有办法提高性能？

import os as s
import numpy as np
import scipy
from PIL import Image
from scipy import ndimage
import warnings
warnings.filterwarnings('ignore')
def imaget_to_pxl(path,size):
    img_to_pxls=np.zeros(shape=(1,size*size*3))
    img_name={'name':0}
    N=1
    #print img_to_pxls.shape
    for i in s.listdir(path):
        if i.endswith('.jpg'):
            new_size=(size,size)
            im = Image.open(path+i, 'r')#.resize(new_size)
            pixel_values = list(im.getdata())
            if im.mode == 'RGB':
                my_image = scipy.misc.imresize(im, size=new_size)#.reshape((1,-1))
                img_to_pxls=np.concatenate((img_to_pxls,my_image.reshape((1,-1))))

    return img_to_pxls

Answer 1

您似乎尝试构建一个在每次迭代中都在增长的 numpy-array。这是您应该不惜一切代价避免的事情！

img_to_pxls=np.zeros(shape=(1,size*size*3))
    # later loop
    my_image = scipy.misc.imresize(im, size=new_size)#.reshape((1,-1))
    img_to_pxls=np.concatenate((img_to_pxls,my_image.reshape((1,-1))))  # !

请注意，调整 numpy-array 的大小会影响分配新数组（增加大小）并复制所有值！这将在数​​组越来越大的每次迭代中发生。

将img_to_pxls 设为一个python 列表（其行为不同！），添加新条目并在收集所有图像后构建一个numpy 数组。

类似的东西（未经测试）：

img_to_pxls = []
...
    my_image = scipy.misc.imresize(im, size=new_size)#.reshape((1,-1))
    img_to_pxls.append(my_image)
...
return np.array(img_to_pxls)

在尝试优化任何内容之前，请始终使用一些分析工具来检查代码中的慢速部分。例如使用line_profiler，您会立即发现这个瓶颈。