Minkowski engine代码

[pypi-image]: https://badge.fury.io/py/MinkowskiEngine.svg [pypi-url]: https://pypi.org/project/MinkowskiEngine/ [pypi-download]: https://img.shields.io/pypi/dm/MinkowskiEngine [slack-badge]: https://img.shields.io/badge/slack-join%20chats-brightgreen [slack-url]: https://join.slack.com/t/minkowskiengine/shared_invite/zt-piq2x02a-31dOPocLt6bRqOGY3U_9Sw # Minkowski Engine [![PyPI Version][pypi-image]][pypi-url] [![pypi monthly download][pypi-download]][pypi-url] [![slack chat][slack-badge]][slack-url] The Minkowski Engine is an auto-differentiation library for sparse tensors. It supports all standard neural network layers such as convolution, pooling, unpooling, and broadcasting operations for sparse tensors. For more information, please visit [the documentation page](http://nvidia.github.io/MinkowskiEngine/overview.html). ## News - 2021-08-11 Docker installation instruction added - 2021-08-06 All installation errors with pytorch 1.8 and 1.9 have been resolved. - 2021-04-08 Due to recent errors in [pytorch 1.8 + CUDA 11](https://github.com/NVIDIA/MinkowskiEngine/issues/330), it is recommended to use [anaconda for installation](#anaconda). - 2020-12-24 v0.5 is now available! The new version provides CUDA accelerations for all coordinate management functions. ## Example Networks The Minkowski Engine supports various functions that can be built on a sparse tensor. We list a few popular network architectures and applications here. To run the examples, please install the package and run the command in the package root directory. | Examples | Networks and Commands | |:---------------------:|:-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:| | Semantic Segmentation | <img src="https://nvidia.github.io/MinkowskiEngine/_images/segmentation_3d_net.png"> <br /> <img src="https://nvidia.github.io/MinkowskiEngine/_images/segmentation.png" width="256"> <br /> `python -m examples.indoor` | | Classification |  <br /> `python -m examples.classification_modelnet40` | | Reconstruction | <img src="https://nvidia.github.io/MinkowskiEngine/_images/generative_3d_net.png"> <br /> <img src="https://nvidia.github.io/MinkowskiEngine/_images/generative_3d_results.gif" width="256"> <br /> `python -m examples.reconstruction` | | Completion | <img src="https://nvidia.github.io/MinkowskiEngine/_images/completion_3d_net.png"> <br /> `python -m examples.completion` | | Detection | <img src="https://nvidia.github.io/MinkowskiEngine/_images/detection_3d_net.png"> | ## Sparse Tensor Networks: Neural Networks for Spatially Sparse Tensors Compressing a neural network to speedup inference and minimize memory footprint has been studied widely. One of the popular techniques for model compression is pruning the weights in convnets, is also known as [*sparse convolutional networks*](https://www.cv-foundation.org/openaccess/content_cvpr_2015/papers/Liu_Sparse_Convolutional_Neural_2015_CVPR_paper.pdf). Such parameter-space sparsity used for model compression compresses networks that operate on dense tensors and all intermediate activations of these networks are also dense tensors. However, in this work, we focus on [*spatially* sparse data](https://arxiv.org/abs/1409.6070), in particular, spatially sparse high-dimensional inputs and 3D data and convolution on the surface of 3D objects, first proposed in [Siggraph'17](https://wang-ps.github.io/O-CNN.html). We can also represent these data as sparse tensors, and these sparse tensors are commonplace in high-dimensional problems such as 3D perception, registration, and statistical data. We define neural networks specialized for these inputs as *sparse tensor networks* and these sparse tensor networks process and generate sparse tensors as outputs. To construct a sparse tensor network, we build all standard neural network layers such as MLPs, non-linearities, convolution, normalizations, pooling operations as the same way we define them on a dense tensor and implemented in the Minkowski Engine. We visualized a sparse tensor network operation on a sparse tensor, convolution, below. The convolution layer on a sparse tensor works similarly to that on a dense tensor. However, on a sparse tensor, we compute convolution outputs on a few specified points which we can control in the [generalized convolution](https://nvidia.github.io/MinkowskiEngine/sparse_tensor_network.html). For more information, please visit [the documentation page on sparse tensor networks](https://nvidia.github.io/MinkowskiEngine/sparse_tensor_network.html) and [the terminology page](https://nvidia.github.io/MinkowskiEngine/terminology.html). | Dense Tensor | Sparse Tensor | |:---------------------------------------------------------------------------:|:----------------------------------------------------------------------------:| | <img src="https://nvidia.github.io/MinkowskiEngine/_images/conv_dense.gif"> | <img src="https://nvidia.github.io/MinkowskiEngine/_images/conv_sparse.gif"> | -------------------------------------------------------------------------------- ## Features - Unlimited high-dimensional sparse tensor support - All standard neural network layers (Convolution, Pooling, Broadcast, etc.) - Dynamic computation graph - Custom kernel shapes - Multi-GPU training - Multi-threaded kernel map - Multi-threaded compilation - Highly-optimized GPU kernels ## Requirements - Ubuntu >= 14.04 - CUDA >= 10.1.243 and **the same CUDA version used for pytorch** (e.g. if you use conda cudatoolkit=11.1, use CUDA=11.1 for MinkowskiEngine compilation) - pytorch >= 1.7 To specify CUDA version, please use conda for installation. You must match the CUDA version pytorch uses and CUDA version used for Minkowski Engine installation. `conda install -y -c nvidia -c pytorch pytorch=1.8.1 cudatoolkit=10.2`) - python >= 3.6 - ninja (for installation) - GCC >= 7.4.0 ## Installation You can install the Minkowski Engine with `pip`, with anaconda, or on the system directly. If you experience issues installing the package, please checkout the [the installation wiki page](https://github.com/NVIDIA/MinkowskiEngine/wiki/Installation). If you cannot find a relevant problem, please report the issue on [the github issue page](https://github.com/NVIDIA/MinkowskiEngine/issues). - [PIP](https://github.com/NVIDIA/MinkowskiEngine#pip) installation - [Conda](https://github.com/NVIDIA/MinkowskiEngine#anaconda) installation - [Python](https://github.com/NVIDIA/MinkowskiEngine#system-python) installation - [Docker](https://github.com/NVIDIA/MinkowskiEngine#docker) installation ### Pip The MinkowskiEngine is distributed via [PyPI MinkowskiEngine][pypi-url] which can be installed simply with `pip`. First, install pytorch following the [instruction](https://pytorch.org). Next, install `openblas`. ``` sudo apt install build-essential python3-dev libopenblas-dev pip install torch ninja pip install -U MinkowskiEngine --install-option="--blas=openblas" -v --no-deps # For pip installation from the latest source # pip install -U git+https://github.com/NVIDIA/MinkowskiEngine --no-deps ``` If you want to specify arguments for the setup script, please refer to the following command. ``` # Uncomment some options if things don't work # export CXX=c++; # set this if you want to u