haiduo / sparse_model

比较了目前主流的3D点云稀疏加速四大框架,具体分为, mit-han-lab的torchsparse、NVIDAI的MinkowskiEngine、 Tusimple的spconv、facebookresearch的SparseConvNet。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

sparse_model

目前较为流行的3D点云稀疏加速四大框架,具体分为, mit-han-lab的torchsparseNVIDAI的MinkowskiEngineTusimple的spconvfacebookresearch的SparseConvNet,具体进行实验以Torchsparse为研究起始,但发现其仅仅是完成了submanifold的实现且功能也不完整(还不成熟),随后经过对spconv的深入研究,发现若要直接的在pytorch层进行修改优化以达到模型稀疏加速的目的是非常有限的,需要在CUDA底层的优化设计进行考虑。

  • 该库主要是对上述的模型稀疏框架进行了配置,消除了安装的错误以及如何解决。

  • 也在VoxelNetSecond上进行了试验。

About

比较了目前主流的3D点云稀疏加速四大框架,具体分为, mit-han-lab的torchsparse、NVIDAI的MinkowskiEngine、 Tusimple的spconv、facebookresearch的SparseConvNet。


Languages

Language:Python 72.2%Language:C++ 14.6%Language:Cuda 4.2%Language:Jupyter Notebook 4.0%Language:JavaScript 3.1%Language:HTML 0.6%Language:PowerShell 0.2%Language:CMake 0.2%Language:C 0.2%Language:Cython 0.2%Language:Dockerfile 0.1%Language:Shell 0.1%Language:CSS 0.0%