閱讀在keras

Question

多個圖像我試圖keras（theano）讀取多個圖像，這樣

import numpy as np 

    import os, sys 

    import cv2 

    my_path='data/train' 

    dirs = os.listdir(my_path) 

    numpy_int_array=np.array(dirs) 

    for i in numpy_int_array: 

     input_im=cv2.imread(numpy_int_array[i],1) 

     print input_im.shape 

我得到以下錯誤，

IndexError：只有整數，切片（:） ，省略號（...）， numpy.newaxis（None）和整數或布爾數組都是有效索引

請幫我在我做錯的地方。

Answer 1

你太很快轉化爲np.array：

for im_name in os.listdir(my_path): # no need to convert to np.array yet... 
    input_im=cv2.imread(im_name, 1) # now it's np.array 
    print input_im.shape 

Python的os.listdir()回報文件列表名（字符串），而不是整數。