hova88 / PointPillars_MultiHead_40FPS

A REAL-TIME 3D detection network [Pointpillars] compiled by CUDA/TensorRT/C++.
GNU General Public License v3.0
279 stars 88 forks source link

Which platform does it run on? X86 or arm? #25

Open YongYongLi123 opened 3 years ago

hova88 commented 3 years ago

both

speshowBUAA commented 3 years ago

arm平台使用是什么板子?我在jetson Nx上实验tensort要跑250ms左右

speshowBUAA commented 3 years ago

@hova88

hova88 commented 3 years ago

arm平台使用是什么板子?我在jetson Nx上实验tensort要跑250ms左右

你好,我用的是agx , 这个工程只是一个很简易的模板,里面还有很多可以优化的地方,需要结合自己使用场景,去掉那些不必要的操作,简化一些设置,基本都到达到使用需求。

speshowBUAA commented 2 years ago

@hova88 请问有没有尝试过在OpenPCDet里更改cbgs_pp_multihead.yaml和nuscenes_dataset.yaml中的点云范围参数 POINT_CLOUD_RANGE: [-11.2, -11.2, -5.0, 11.2, 11.2, 3.0],来达到压缩模型大小的效果?我尝试后训练结果出现了异常

hova88 commented 2 years ago

@hova88 请问有没有尝试过在OpenPCDet里更改cbgs_pp_multihead.yaml和nuscenes_dataset.yaml中的点云范围参数 POINT_CLOUD_RANGE: [-11.2, -11.2, -5.0, 11.2, 11.2, 3.0],来达到压缩模型大小的效果?我尝试后训练结果出现了异常

异常大概率是因为那几个cuda kernel写的太固定了,回头有时间我给改成dynamic shape的方式