pointnet shuffle_data(data, labels)

本文介绍了一种用于打乱点云数据及其标签顺序的方法,通过使用numpy库的shuffle函数和自定义函数shuffle_data,实现了对数据和标签的随机化处理,这对于训练机器学习模型时避免数据顺序对结果的影响至关重要。

打乱point点云元数据顺序以及labels的顺序

def shuffle_data(data, labels):
    """ Shuffle data and labels.
        Input:
          data: B,N,... numpy array
          label: B,... numpy array
        Return:
          shuffled data, label and shuffle indices
    """
    idx = np.arange(len(labels))
    np.random.shuffle(idx)
    return data[idx, ...], labels[idx], idx

调用语句为

current_data, current_label, _ = provider.shuffle_data(current_data, np.squeeze(current_label))            
        

np.squeeze删除current_label中单维数据,只留filesize的那一维数据,因为h5文件中存储的point的模型数目并非是与batchsize完全匹配的,这个数后续还会有处理

经过将filesize维度数据顺序打乱,实现随机打乱current_label以及current_data数据的目的

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值