3D点云算法PointNet++中最远点采样源码实现及详解

本文详细介绍了PointNet++中采用的最远点采样(Farthest Point Sampling, FPS)方法,用于特征提取。该方法通过不断选取当前批次中与已选点最远的点作为新的采样点,直至达到预设的采样点数。在给定的代码示例中,可以看到FPS的实现过程,包括随机选取初始点,计算距离,更新距离矩阵,并找出新的最远点,不断迭代完成采样。这种方法在3D点云处理中起到关键作用,确保了采样的代表性。

在PointNet++中,特征提取的采样点是通过最远点采样法得到的。其基本思想就是:

①首先随机取一个点a,然后遍历其余剩余的点,计算与a的距离,得到最远距离点b;

②遍历剩余点与a,b的距离,得到距离最远的点c;

③重复执行以上,直到取够设定点的个数。

具体可举例,下图展示的是一个batch中的计算过程,程序中8个batch同时进行的。

 

 ①随机选取中心点A,建立距离分布图distance并赋很大的值;

②分别计算A点到B、C、D、E点距离,得到A_Dis;将A_Dis与distance比较,如果Dis中存在小于distance中的值,就将其更新到distance中;然后根据新的distance图,获得最大距离值的点D;

③以D点为中心点,计算与其他点的位置,得到D_Dis;将D_Dis与distance比较,如果Dis中存在小于distance中的值,就将其更新到distance中;然后根据新的distance图,获得最大距离值的点E;

④重复以上循环,得到中心点B,C。

其程序及详解可参考源码。

def farthest_point_sample(xyz, npoint):
    """
    Input:
        xyz: pointcloud data, [B, N, 3],如batch=8,输入点N=1024,位置信息xyz=3
        npoint: numb
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值