在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()返回文件名称（字符串）的列表，而不是整数。