method2- list
def load(data_dir, label_file, test=False): image_size = 96 col_num = 1+68*2 #image_list_np = np.array([]) label_list = [] image_list = [] with open(label_file, 'rb') as csv_file: all_lines = csv.reader(csv_file, delimiter=" ", skipinitialspace=True) for i,row in enumerate(all_lines): if not len(row) == col_num: continue label_list.append(row[1:]) image_path = os.path.join(data_dir,row[0]) im = cv2.imread(image_path) im = cv2.resize(im, (image_size,image_size)) im_list = im.tolist() image_list.append(im_list) ''' if i == 0: image_list_np = np.expand_dims(im, axis=0) else: expend_im = np.expand_dims(im, axis=0) #image_list_np = np.concatenate((image_list_np, expend_im)) image_list_np = np.append(image_list_np, expend_im, axis=0) ''' label_list_np = np.asarray(label_list, dtype=np.float32) image_list_np= np.asarray(image_list, dtype=np.float32) return image_list_np, label_list_np