读取批量图片向量以及标签

发现用的比较多，就直接写下来，下次直接改路径调用

import cv2
import os
import numpy as np
from keras.utils.np_utils import to_categorical

data1_path = '/home/lu/lw/gan/ppt'
class_folders = os.listdir(data1_path)

num_classes = len(class_folders)
train_x = []
train_y = []
#test_x = []
#test_y = []
img_size = 128


train_files = os.listdir(data1_path)
train_label_path = '/home/lu/lw/gan/pptrain.txt'

train_lines = open(train_label_path).readlines()

for line in train_lines[0:]:
	tmp_label = line.split(' ')[1]
	train_y.append(tmp_label)

for img_file in train_files:
	if img_file.split('.')[-1] == 'jpg':
		img = cv2.imread(data1_path+'/'+img_file)
		img = cv2.resize(img,(img_size, img_size))
		train_x.append(img)#

train_x = np.array(train_x)

#train_y = to_categorical(train_y)
#print train_x.shape[0]
print train_x
print train_y 
print len(train_x[0])
print len(train_y[0])