def read_img(path):
imgs = []
labels = []
# classs = []
classs = os.listdir(path)
for idx, folder in enumerate(classs):
cate = os.path.join(path, folder)
for im in os.listdir(cate):
print(im)
img_path =os.path.join(cate, im)
print('reading the images:%s' % (img_path))
img = io.imread(img_path)
img = transform.resize(img, (w, h))
# with open('tests.txt', 'a') as f:
# f.write(img_path+'_'+str(idx)+'\n')
imgs.append(img)
labels.append(idx)
return np.asarray(imgs, np.float32), np.asarray(labels, np.int32)
