You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+2-7Lines changed: 2 additions & 7 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,8 +1,8 @@
1
-
# torchsparse: High-Performance Computing Library for Efficient 3D Sparse Convolution
1
+
# TorchSparse
2
2
3
3
## News
4
4
5
-
[**New**]2020/09/20: We released `torchsparse` v1.1, which is significantly faster than our `torchspare` v1.0 and is also achieves **1.9x** speedup over [MinkowskiEngine](https://github.com/NVIDIA/MinkowskiEngine) v0.5 alpha when running MinkUNet18C!
5
+
2020/09/20: We released `torchsparse` v1.1, which is significantly faster than our `torchspare` v1.0 and is also achieves **1.9x** speedup over [MinkowskiEngine](https://github.com/NVIDIA/MinkowskiEngine) v0.5 alpha when running MinkUNet18C!
6
6
7
7
2020/08/30: We released `torchsparse` v1.0.
8
8
@@ -87,8 +87,6 @@ where we need to define `inputs`(SparseTensor), `kernel` (of shape k^3 x OC x IC
87
87
88
88
Detailed examples are given in [here](https://github.com/mit-han-lab/e3d/blob/master/spvnas/core/modules/dynamic_sparseop.py), where we use the `torchsparse.nn.functional` interfaces to implement weight-shared 3D-NAS modules.
89
89
90
-
91
-
92
90
### Sparse Hashmap API
93
91
94
92
Sparse hash map query is important in 3D sparse computation. It is mainly used to infer a point's memory location (*i.e.* index) given its coordinates. For example, we use this operation in kernel map construction part of 3D sparse convolution, and also sparse voxelization / devoxelization in [Sparse Point-Voxel Convolution](https://arxiv.org/abs/2007.16100). Here, we provide the following example for hash map API:
In this example, `sphash` is the function converting integer coordinates to hashing. The `sphashquery(source_hash, target_hash)` performs the hash table lookup. Here, the hash map has key `target_hash` and value corresponding to point indices in the target point cloud tensor. For each point in the `source_coords`, we find the point index in `target_coords` which has the same coordinate as it.
103
101
104
-
105
-
106
102
### Dummy Training Example
107
103
108
104
We here provides an entire training example with dummy input [here](examples/example.py). In this example, we cover
@@ -114,7 +110,6 @@ We here provides an entire training example with dummy input [here](examples/exa
114
110
You are also welcomed to check out our [SPVNAS](https://github.com/mit-han-lab/e3d) project to implement training / inference with real data.
115
111
116
112
117
-
118
113
## Speed Comparison Between torchsparse and MinkowskiEngine
119
114
120
115
We benchmark the performance of our torchsparse and latest [MinkowskiEngine V0.4.3](https://github.com/NVIDIA/MinkowskiEngine) here, latency is measured on NVIDIA GTX 1080Ti GPU:
0 commit comments