numpy array append or concatenate

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值