torchsparse:用于点云处理的高性能神经网络库 源码

火炬稀疏 消息 2020/09/20:我们发布了torchsparse v1.1,该版本明显快于我们的torchsparse v1.0,并且在运行MinkUNet18C时,其速度比 v0.5 alpha快1.9倍! 2020/08/30:我们发布了torchsparse v1.0。 概述 我们发布了torchsparse ,这是一个用于高效3D稀疏卷积的高性能计算库。 该库旨在加速3D稀疏计算,尤其是稀疏卷积运算。 该库的主要优点是我们支持GPU上的所有计算,尤其是内核映射构造(这是在最新的 V0.4.3中的CPU上完成的)。 安装 您可以运行以下命令来安装torchsparse。 pip install --upgrade git+https://github.com/mit-han-lab/torchsparse.git 请注意,该库取决于Google的。 为了安装该库,您可以