点云深度学习系列5——pointnet++文章及代码分析

大家好。

PointNet++是PointNet的升级版本,增加了对局部信息的感知能力。体现到代码上的话,变化还是比较多的,我们以分类为例,对结构和代码进行分析。

网络结构

 

首先是网络结构方面,复习前任PointNet网络结构的,请点这里

改进版去掉了T-net,在网络层次上变多了,但是更加组织有序。

 

def get_model(point_cloud, is_training, bn_decay=None):  
    """ Classification PointNet, input is BxNx3, output Bx40 """  
    batch_size = point_cloud.get_shape()[0].value  
    num_point = point_cloud.get_shape()[1].value  
    end_points = {}  
    l0_xyz = point_cloud  
    l0_points = None  
    end_points['l0_xyz'] = l0_xyz  

    # Set abstraction layers  
    # Note: When using NCHW for layer 2, we see increased GPU memory usage (in TF1.4).  
    # So we only use NCHW for layer 1 until this issue can be resolved.  
    l1_xyz, l1_points, l1_indices = pointnet_sa_module(l0_xyz, l0_points, npoint=512, radius=0.2, nsample=32, mlp=[64,64,128], mlp2=None, group_all=False, is_training=is_training, bn_decay=bn_decay, scope='layer1', use_nchw=True)  
    l2_xyz, l2_points, l2_indices = pointnet_sa_module(l1_xyz, l1_points, npoint=128, radius=0.4, nsample=64, mlp=[
  • 2
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 12
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值