# 如何得到train_x和Train_y import os from keras.preprocessing.image import ImageDataGenerator, load_img, img_to_array from PIL import Image import numpy as np ''' ''' def load_data(): print('load data from image') imgs = os.listdir("../data/train/") num = len(imgs) print("num", num) data = np.empty((num, 1, 50, 28), dtype="float32") label = np.empty((num,), dtype="uint8") i = 0 for imgFile in imgs: img = Image.open("../data/train/" + imgFile).convert('L') arr = np.asarray(img, dtype="float32") data[i, :, :, :] = arr label[i] = int(imgs[i].split('.')[0].split('_')[1]) i = i + 1 return data, label if __name__ == "__main__": data, label = load_data() for i in label: print("label:", i)
从图片得到keras的训练数据
猜你喜欢
转载自my.oschina.net/payzheng/blog/1647942
今日推荐
周排行