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