ocnn 2.2.5__py3-none-any.whl → 2.2.6__py3-none-any.whl

ocnn/__init__.py

@@ -12,7 +12,7 @@ from . import models
 from . import dataset
 from . import utils
 
-__version__ = '2.2.5'
+__version__ = '2.2.6'
 
 __all__ = [
     'octree',

ocnn/octree/octree.py

@@ -153,7 +153,9 @@ class Octree:
       point_cloud (Points): The input point cloud.
 
     .. note::
-      Currently, the batch size of the point cloud must be 1.
+      The point cloud must be strictly in range :obj:`[-1, 1]`. A good practice
+      is to normalize it into :obj:`[-0.99, 0.99]` or :obj:`[0.9, 0.9]` to retain
+      some margin.
     '''
 
     self.device = point_cloud.device

ocnn/octree/points.py

@@ -63,9 +63,9 @@ class Points:
     if self.batch_id is not None:
       assert self.batch_id.dim() == 2 or self.batch_id.dim() == 1
       assert self.batch_id.size(0) == self.points.size(0)
-      assert self.batch_id.size(1) == 1
       if self.batch_id.dim() == 1:
         self.batch_id = self.batch_id.unsqueeze(1)
+      assert self.batch_id.size(1) == 1
 
   @property
   def npt(self):

{ocnn-2.2.5.dist-info → ocnn-2.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ocnn
-Version: 2.2.5
+Version: 2.2.6
 Summary: Octree-based Sparse Convolutional Neural Networks
 Home-page: https://github.com/octree-nn/ocnn-pytorch
 Author: Peng-Shuai Wang
@@ -28,30 +28,41 @@ Requires-Dist: packaging
 
 This repository contains the **pure PyTorch**-based implementation of
 [O-CNN](https://wang-ps.github.io/O-CNN.html). The code has been tested with
-`Pytorch>=1.6.0`, and `Pytorch>=1.9.0` is preferred.
+`Pytorch>=1.6.0`, and `Pytorch>=1.9.0` is preferred. The *original*
+implementation of O-CNN is based on C++ and CUDA and can be found
+[here](https://github.com/Microsoft/O-CNN), which has received
+[![stars - O-CNN](https://img.shields.io/github/stars/microsoft/O-CNN?style=social)](https://github.com/microsoft/O-CNN) and
+[![forks - O-CNN](https://img.shields.io/github/forks/microsoft/O-CNN?style=social)](https://github.com/microsoft/O-CNN).
 
-O-CNN is an octree-based sparse convolutional neural network framework for 3D
-deep learning. O-CNN constrains the CNN storage and computation into non-empty
-sparse voxels for efficiency and uses the `octree` data structure to organize
-and index these sparse voxels.
 
-The concept of sparse convolution in O-CNN is the same with
-[H-CNN](https://ieeexplore.ieee.org/abstract/document/8580422),
+O-CNN is an octree-based 3D convolutional neural network framework for 3D data.
+O-CNN constrains the CNN storage and computation into non-empty sparse voxels
+for efficiency and uses the `octree` data structure to organize and index these
+sparse voxels. Currently, this type of 3D convolution is known as Sparse
+Convolution in the research community.
+
+
+The concept of Sparse Convolution in O-CNN is the same with
 [SparseConvNet](https://openaccess.thecvf.com/content_cvpr_2018/papers/Graham_3D_Semantic_Segmentation_CVPR_2018_paper.pdf),
-and
-[MinkowskiNet](https://openaccess.thecvf.com/content_CVPR_2019/papers/Choy_4D_Spatio-Temporal_ConvNets_Minkowski_Convolutional_Neural_Networks_CVPR_2019_paper.pdf).
-The key difference is that our O-CNN uses the `octree` to index the sparse
-voxels, while these 3 works use the `Hash Table`.
-
-Our O-CNN is published in SIGGRAPH 2017, H-CNN is published in TVCG 2018,
-SparseConvNet is published in CVPR 2018, and MinkowskiNet is published in
-CVPR 2019. Actually, our O-CNN was submitted to SIGGRAPH in the end of 2016 and
-was officially accepted in March, 2017. The camera-ready version of our O-CNN was
-submitted to SIGGRAPH in April, 2017. We just did not post our paper on Arxiv
-during the review process of SIGGRAPH. Therefore, **the idea of constraining CNN
-computation into sparse non-emtpry voxels is first proposed by our O-CNN**.
-Currently, this type of 3D convolution is known as Sparse Convolution in the
-research community.
+[MinkowskiNet](https://github.com/NVIDIA/MinkowskiEngine), and
+[SpConv](https://github.com/traveller59/spconv).
+The key difference is that our O-CNN uses `octrees` to index the sparse voxels,
+while these works use `Hash Tables`. However, I believe that `octrees` may be
+the right choice for Sparse Convolution. With `octrees`, I can implement the
+Sparse Convolution with pure PyTorch. More importantly, with `octrees`, I can
+also build efficient transformers for 3D data --
+[OctFormer](https://github.com/octree-nn/octformer), which is extremely hard
+with `Hash Tables`.
+
+
+Our O-CNN is published in SIGGRAPH 2017, SparseConvNet is published in CVPR
+2018, and MinkowskiNet is published in CVPR 2019. Actually, our O-CNN was
+submitted to SIGGRAPH in the end of 2016 and was officially accepted in March,
+2017. <!-- The camera-ready version of our O-CNN was submitted to SIGGRAPH in April, 2018. -->
+We just did not post our paper on Arxiv during the review process of SIGGRAPH.
+Therefore, **the idea of constraining CNN computation into sparse non-emtpry
+voxels, i.e. Sparse Convolution,  is first proposed by our O-CNN**.
+
 
 ## Key benefits of ocnn-pytorch
 

{ocnn-2.2.5.dist-info → ocnn-2.2.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-ocnn/__init__.py,sha256=z0LmCb7hdmNZekWucdzmEvhYse0eJ5L3jGScTreJ7ok,582
+ocnn/__init__.py,sha256=Vek3w8YFUUmHO2aGzIHBD_ZUeKzWyJxAPS0ZDK1bUUM,582
 ocnn/dataset.py,sha256=wvclvjlZs9qTeMXWLaO32K5d1VVY9XHSNuVVJEpVeeo,5266
 ocnn/utils.py,sha256=XhykveOjHoQd94gjJ5-opzXs-9MOCAzZ34ArZ8mG4sE,6726
 ocnn/models/__init__.py,sha256=F9PJRhOPHc1OrwkqcfywEBW0J6jmVW7-IHgWjGpY15U,724
@@ -26,11 +26,11 @@ ocnn/nn/octree_norm.py,sha256=XrSQ7oZKfepYqQIuPU4loaYNae_rgtPpekR0cqTvryg,4524
 ocnn/nn/octree_pad.py,sha256=suV6Ftb-UlUuoJzKdQ9DCP9oqVQJq7vXb9_6hq-kUk4,1323
 ocnn/nn/octree_pool.py,sha256=Zn2XLk5SFl6pqMhhKIvu1uZl5ebonFSlPcKr54fOIPA,6664
 ocnn/octree/__init__.py,sha256=vKZFc5_r6Gxg5KsPWiCZCR-umWWfWPoE7qBx4PIrUGA,630
-ocnn/octree/octree.py,sha256=SNhzI-UOce2oPtQ_KLmR2lduY2P1_ne5fe_4BLEhlv0,24590
-ocnn/octree/points.py,sha256=7_iiN0y0g9SgV8kLzuGW20C-MFOmz5hh1aQhDhUIa0Q,11355
+ocnn/octree/octree.py,sha256=uFv0nNR_lIh6SgpDjlj4lraRbUolo2_gdfGVvyP5RxA,24711
+ocnn/octree/points.py,sha256=TLlUGdguTkLDRxuEhC-1ejik6thuAWX3L-UiONbPLeM,11355
 ocnn/octree/shuffled_key.py,sha256=UJZ4eKNA_7nLbf9FbEvS_3VyrAqnZCOzk1hsPtJianM,3936
-ocnn-2.2.5.dist-info/LICENSE,sha256=YeOS0Plo8Uistv_8ZXdgddmN9GHJKnIiJ5FZ8zTW6Sw,1114
-ocnn-2.2.5.dist-info/METADATA,sha256=1r105xmAHwGG9tYy-xcnXuLCZCWVH_jrtao8wLZgCtk,3773
-ocnn-2.2.5.dist-info/WHEEL,sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY,91
-ocnn-2.2.5.dist-info/top_level.txt,sha256=ayZdVOnxlOke3kgzAlrRh2IEL_qOudwOaEU3xhjtpZ0,5
-ocnn-2.2.5.dist-info/RECORD,,
+ocnn-2.2.6.dist-info/LICENSE,sha256=YeOS0Plo8Uistv_8ZXdgddmN9GHJKnIiJ5FZ8zTW6Sw,1114
+ocnn-2.2.6.dist-info/METADATA,sha256=pMj-GJiTUtxvjtjhBLQcBMi1CXZlU3NBiQKxHiPJR8M,4382
+ocnn-2.2.6.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+ocnn-2.2.6.dist-info/top_level.txt,sha256=ayZdVOnxlOke3kgzAlrRh2IEL_qOudwOaEU3xhjtpZ0,5
+ocnn-2.2.6.dist-info/RECORD,,

{ocnn-2.2.5.dist-info → ocnn-2.2.6.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.5.0)
+Generator: setuptools (75.6.0)
 Root-Is-Purelib: true
 Tag: py3-none-any