파이썬을 사용하여 이미지 행렬

Question

폴더의 모든 이미지에 액세스하여 매트릭스에 저장해야합니다. 내가 MATLAB을 사용하여 할 수 있었고, 여기에 코드입니다 :

input_dir = 'C:\Users\Karim\Downloads\att_faces\New Folder'; 
image_dims = [112, 92]; 

filenames = dir(fullfile(input_dir, '*.pgm')); 
num_images = numel(filenames); 

images = []; 
for n = 1:num_images 
    filename = fullfile(input_dir, filenames(n).name); 
    img = imread(filename); 
    img = imresize(img,image_dims); 
end 

하지만 난 파이썬을 사용하여 작업을 수행하는 데 필요한 여기 내 파이썬 코드입니다 :

import Image 
import os 
from PIL import Image 
from numpy import * 
import numpy as np 


#import images 
dirname = "C:\\Users\\Karim\\Downloads\\att_faces\\New folder" 


#get number of images and dimentions 
path, dirs, files = os.walk(dirname).next() 
num_images = len(files) 
image_file = "C:\\Users\\Karim\\Downloads\\att_faces\\New folder\\2.pgm" 
im = Image.open(image_file) 
width, height = im.size 



images = [] 

for x in xrange(1, num_images): 
    filename = os.listdir(dirname)[x] 
    img = Image.open(filename) 
    img = im.convert('L') 
    images[:, x] = img[:] 

하지만 난이 오류를 얻고있다 : IOError : [Errno 2] 해당 파일이나 디렉터리가 없습니다 : '10 .pgm ' 파일이 있지만.

Answer 1

나는 당신의 최종 목표는 무엇인지 확실히 모르겠지만, 더 이런 식으로 뭔가를 시도 :

import numpy as np 
import Image 
import glob 

filenames = glob.glob('/path/to/your/files/*.pgm') 
images = [Image.open(fn).convert('L') for fn in filenames] 
data = np.dstack([np.array(im) for im in images]) 

이 폭 x 높이 X NUM_IMAGES NumPy와 배열을 얻을 것입니다, 모든 이미지가있는 가정 같은 치수.

그러나 이미지는 정렬되지 않으므로 filenames.sort() 할 수 있습니다.

또한 여러분은 물건을 3D numpy 배열로 원할 수도 있고 원하지 않을 수도 있지만, 실제로는 실제로하고있는 것에 달려 있습니다. 각 "프레임"을 개별적으로 조작하려는 경우 하나의 거대한 배열로 스택하지 않아도됩니다.