image_list_np = np.array([])
for i,row in enumerate(all_lines):
image_path = os.path.join(data_dir,row[0])
im = cv2.imread(image_path)
im = cv2.resize(im, (image_size,image_size))
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)
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