#177 · 主分类: 深度学习框架

MinkowskiEngine

3d-convolutional-network 3d-vision 4d-convolutional-neural-network auto-differentiation computer-vision convolutional-neural-networks cuda deep-learning high-dimensional-data high-dimensional-inference minkowski-engine neural-network pytorch semantic-segmentation space-time sparse-convolution sparse-tensor-network sparse-tensors spatio-temporal-analysis trilateral-filter

Minkowski Engine 是一个用于高维稀疏张量的自动微分神经网络库。

项目最后更新:03/05/24

GitHub Stars

3.0K

Forks数量

484

贡献者数量

25

许可证

Other

收录理由

三维感知任务中,大量输入数据其实是空荡荡的体素空间,MinkowskiEngine 顺应这一点,而不是硬塞进稠密张量。它是一套面向稀疏张量的自动微分库,卷积、池化、反池化、广播等操作都直接作用于真实存在的点,绕开稠密表示被迫填充的空体素,因此处理激光雷达扫描或大型三维场景时,内存占用和训练耗时都能压下来。它嵌在 PyTorch 生态里,提供 CUDA 加速内核,还带室内分割、分类、重建等可直接运行的示例,方便在正式投入项目前先验证行为。主要门槛在安装:需要经历 CUDA 编译,并仔细对齐 PyTorch 版本。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目