c++实现pointpillar的前处理

主要为PillarVFE和PointPillarScatter部分。

流程:

  • 读取bin格式点云

PillarVFE:

  • 根据设定大小体素化,每个体素内存放至多32个点,每个点10个特征(x,y,z,r,等)。体素用结构体存储,Points为(32,10)的数组:
struct Voxel {
    //float x, y, z, height, width, length, intensity;    
    int point_num = 0 ;  //体素内点云数
    float sum_x = 0;     //x坐标求和后平均
    float sum_y = 0;     //y坐标求和后平均
    float sum_z = 0;     //z坐标求和后平均
    std::vector<std::vector<float>> Points;
};
  • PFN,用linear层将10维特征扩展到64维。耗时主要在这一层。权重数据经pytorch保存为txt文件后读取。
  • Maxpool,对比32个点中的每维特征,保留最大值。64->1,降为1维数组

 PointPillarScatter:

         拆分voxel维度(原本是H*W个体素,现在拆开[H,W]),填入一个三维数组中。       

结果:

体素大小 :0.16*0.16*4

  • VOXEL time = 0.138s

  • PFN time = 3.029s

  • max time = 0.263s

  • time = 4.219s

  • 非空体素数目  8377

  • 总体素数目  251001

  • 非空率 8377/251001 = 3.3%

    

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值