#177 · Primary category: Deep Learning Frameworks

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 is an auto-diff neural network library for high-dimensional sparse tensors

Project last updated:03/05/24

GitHub Stars

3.0K

Forks

484

Contributors

25

License

Other

Why we included this project

Most 3D perception work starts with data that is mostly empty space, and MinkowskiEngine is built around that fact instead of fighting it. It is an auto-differentiation library for sparse tensors, so convolution, pooling, unpooling, and broadcasting all operate directly on the points that actually exist, skipping the empty voxels a dense tensor would force you to pad. That keeps memory and training time down for LiDAR scans and large 3D scenes. It slots into PyTorch, ships CUDA-accelerated kernels, and includes runnable examples for indoor segmentation, classification, and reconstruction, which is handy for checking how it behaves before you commit a project to it. The main friction is installation: expect a CUDA build step and some care matching PyTorch versions.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category