ocnn 2.2.2__py3-none-any.whl → 2.2.3__py3-none-any.whl

ocnn/__init__.py

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

ocnn/models/lenet.py

@@ -26,7 +26,7 @@ class LeNet(torch.nn.Module):
     self.convs = torch.nn.ModuleList([ocnn.modules.OctreeConvBnRelu(
         channels[i], channels[i+1], nempty=nempty) for i in range(stages)])
     self.pools = torch.nn.ModuleList([ocnn.nn.OctreeMaxPool(
-        nempty) for i in range(stages)])
+        nempty) for _ in range(stages)])
     self.octree2voxel = ocnn.nn.Octree2Voxel(self.nempty)
     self.header = torch.nn.Sequential(
         torch.nn.Dropout(p=0.5),                     # drop1

ocnn/nn/__init__.py

@@ -15,6 +15,7 @@ from .octree_pool import (octree_max_pool, OctreeMaxPool,
                           octree_global_pool, OctreeGlobalPool,
                           octree_avg_pool, OctreeAvgPool,)
 from .octree_conv import OctreeConv, OctreeDeconv
+from .octree_gconv import OctreeGroupConv
 from .octree_dwconv import OctreeDWConv
 from .octree_norm import OctreeBatchNorm, OctreeGroupNorm, OctreeInstanceNorm
 from .octree_drop import OctreeDropPath
@@ -32,7 +33,7 @@ __all__ = [
     'OctreeMaxPool', 'OctreeMaxUnpool',
     'OctreeGlobalPool', 'OctreeAvgPool',
     'OctreeConv', 'OctreeDeconv',
-    'OctreeDWConv',
+    'OctreeGroupConv', 'OctreeDWConv',
     'OctreeInterp', 'OctreeUpsample',
     'OctreeInstanceNorm', 'OctreeBatchNorm', 'OctreeGroupNorm',
     'OctreeDropPath',

ocnn/nn/octree_conv.py

@@ -109,6 +109,12 @@ class OctreeConvBase:
     r''' Peforms the forward pass of octree-based convolution.
     '''
 
+    # Type check
+    if data.dtype != out.dtype:
+      data = data.to(out.dtype)
+    if weights.dtype != out.dtype:
+      weights = weights.to(out.dtype)
+
     # Initialize the buffer
     buffer = data.new_empty(self.buffer_shape)
 
@@ -139,6 +145,12 @@ class OctreeConvBase:
     r''' Performs the backward pass of octree-based convolution.
     '''
 
+    # Type check
+    if grad.dtype != out.dtype:
+      grad = grad.to(out.dtype)
+    if weights.dtype != out.dtype:
+      weights = weights.to(out.dtype)
+
     # Loop over each sub-matrix
     for i in range(self.buffer_n):
       start = i * self.buffer_h
@@ -165,6 +177,12 @@ class OctreeConvBase:
     r''' Computes the gradient of the weight matrix.
     '''
 
+    # Type check
+    if data.dtype != out.dtype:
+      data = data.to(out.dtype)
+    if grad.dtype != out.dtype:
+      grad = grad.to(out.dtype)
+
     # Record the shape of out
     out_shape = out.shape
     out = out.flatten(0, 1)

ocnn/nn/octree_dwconv.py

@@ -32,6 +32,12 @@ class OctreeDWConvBase(OctreeConvBase):
     r''' Peforms the forward pass of octree-based convolution.
     '''
 
+    # Type check
+    if data.dtype != out.dtype:
+      data = data.to(out.dtype)
+    if weights.dtype != out.dtype:
+      weights = weights.to(out.dtype)
+
     # Initialize the buffer
     buffer = data.new_empty(self.buffer_shape)
 
@@ -62,6 +68,12 @@ class OctreeDWConvBase(OctreeConvBase):
     r''' Performs the backward pass of octree-based convolution. 
     '''
 
+    # Type check
+    if grad.dtype != out.dtype:
+      grad = grad.to(out.dtype)
+    if weights.dtype != out.dtype:
+      weights = weights.to(out.dtype)
+
     # Loop over each sub-matrix
     for i in range(self.buffer_n):
       start = i * self.buffer_h
@@ -88,6 +100,12 @@ class OctreeDWConvBase(OctreeConvBase):
     r''' Computes the gradient of the weight matrix.
     '''
 
+    # Type check
+    if data.dtype != out.dtype:
+      data = data.to(out.dtype)
+    if grad.dtype != out.dtype:
+      grad = grad.to(out.dtype)
+
     # Record the shape of out
     out_shape = out.shape
     out = out.flatten(0, 1)

ocnn/octree/octree.py

@@ -274,7 +274,7 @@ class Octree:
       children[0] = 0
 
     # update octree
-    self.children[depth] = children
+    self.children[depth] = children.int()
     self.nnum_nempty[depth] = nnum_nempty
 
   def octree_grow(self, depth: int, update_neigh: bool = True):
@@ -498,7 +498,7 @@ class Octree:
     # normalize xyz to [-1, 1] since the average points are in range [0, 2^d]
     if rescale:
       scale = 2 ** (1 - depth)
-      xyz = self.points[depth] * scale - 1.0
+      xyz = xyz * scale - 1.0
 
     # construct Points
     out = Points(xyz, self.normals[depth], self.features[depth],

ocnn/octree/points.py

@@ -56,11 +56,16 @@ class Points:
       assert self.features.dim() == 2
       assert self.features.size(0) == self.points.size(0)
     if self.labels is not None:
-      assert self.labels.dim() == 2
+      assert self.labels.dim() == 2 or self.labels.dim() == 1
       assert self.labels.size(0) == self.points.size(0)
+      if self.labels.dim() == 1:
+        self.labels = self.labels.unsqueeze(1)
     if self.batch_id is not None:
-      assert self.batch_id.dim() == 2 and self.batch_id.size(1) == 1
+      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)
 
   @property
   def npt(self):

ocnn/utils.py

@@ -173,7 +173,7 @@ def resize_with_last_val(list_in: list, num: int = 3):
   '''
 
   assert (type(list_in) is list and len(list_in) < num + 1)
-  for i in range(len(list_in), num):
+  for _ in range(len(list_in), num):
     list_in.append(list_in[-1])
   return list_in
 
@@ -186,15 +186,18 @@ def list2str(list_in: list):
   return ''.join(out)
 
 
-def build_example_octree(depth: int = 5, full_depth: int = 2):
-  r''' Builds an example octree on CPU from 3 points.
+def build_example_octree(depth: int = 5, full_depth: int = 2, pt_num: int = 3):
+  r''' Builds an example octree on CPU from at most 3 points.
   '''
   # initialize the point cloud
   points = torch.Tensor([[-1, -1, -1], [0, 0, -1], [0.0625, 0.0625, -1]])
   normals = torch.Tensor([[1, 0, 0], [-1, 0, 0], [0, 1, 0]])
   features = torch.Tensor([[1, -1], [2, -2], [3, -3]])
   labels = torch.Tensor([[0], [2], [2]])
-  point_cloud = ocnn.octree.Points(points, normals, features, labels)
+
+  assert pt_num <= 3 and pt_num > 0
+  point_cloud = ocnn.octree.Points(
+      points[:pt_num], normals[:pt_num], features[:pt_num], labels[:pt_num])
 
   # build octree
   octree = ocnn.octree.Octree(depth, full_depth)

{ocnn-2.2.2.dist-info → ocnn-2.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ocnn
-Version: 2.2.2
+Version: 2.2.3
 Summary: Octree-based Sparse Convolutional Neural Networks
 Home-page: https://github.com/octree-nn/ocnn-pytorch
 Author: Peng-Shuai Wang
@@ -43,14 +43,14 @@ 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 
+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. 
+research community.
 
 ## Key benefits of ocnn-pytorch
 
@@ -65,3 +65,15 @@ research community.
   training settings, MinkowskiNet 0.4.3 takes 60 hours and MinkowskiNet 0.5.4
   takes 30 hours.
 
+## Citation
+
+  ```bibtex
+  @article {Wang-2017-ocnn,
+    title    = {{O-CNN}: Octree-based Convolutional Neural Networksfor {3D} Shape Analysis},
+    author   = {Wang, Peng-Shuai and Liu, Yang and Guo, Yu-Xiao and Sun, Chun-Yu and Tong, Xin},
+    journal  = {ACM Transactions on Graphics (SIGGRAPH)},
+    volume   = {36},
+    number   = {4},
+    year     = {2017},
+  }
+  ```

{ocnn-2.2.2.dist-info → ocnn-2.2.3.dist-info}/RECORD

@@ -1,11 +1,11 @@
-ocnn/__init__.py,sha256=3vM-r35p3glXfQ00Pqe13Loig8b8zk17l-_lw2npsP0,582
+ocnn/__init__.py,sha256=kCGigMn_30MohB49Hwy0ZBXf5HTUA_i-QOjckrM48Nc,582
 ocnn/dataset.py,sha256=wvclvjlZs9qTeMXWLaO32K5d1VVY9XHSNuVVJEpVeeo,5266
-ocnn/utils.py,sha256=w6IrVsBMAL5JuBUxg1BtcuWkCpy8Fo5LkYG6arYpFkc,6618
+ocnn/utils.py,sha256=XhykveOjHoQd94gjJ5-opzXs-9MOCAzZ34ArZ8mG4sE,6726
 ocnn/models/__init__.py,sha256=F9PJRhOPHc1OrwkqcfywEBW0J6jmVW7-IHgWjGpY15U,724
 ocnn/models/autoencoder.py,sha256=TjOet3dbZLexz-PvZdeV0mbIuGpUAeSH6KWU_z_s-d8,5906
 ocnn/models/hrnet.py,sha256=9W2fi7Fuw0JXDBiZtEoUW2K7ghtpWUm_BWd-mKoHLY0,6684
 ocnn/models/image2shape.py,sha256=5djcOHJh2SQCwd5XdLPeL5vlQDNWnRU3tJY3ojRI8aQ,4589
-ocnn/models/lenet.py,sha256=oryTVauF48biFIJY163d15JNTYHU4M_gBHUWKDM9n7Q,1754
+ocnn/models/lenet.py,sha256=ujVBxnn8AKiIKqB4WHxuK728oJe4TZdqZUTTNy8y3zE,1754
 ocnn/models/ounet.py,sha256=sGGVHqF1nKw798vyththxp7-P3wHlsDUO39hbQpUVTw,3303
 ocnn/models/resnet.py,sha256=9gZKbhFituqPJHCm-bp1xTtHKZ6wu77s31a0q7RtXoQ,2029
 ocnn/models/segnet.py,sha256=VfZf8gBMSPgO5m8Agsfccw9snTrI4LGh0SqxhLZ1F8s,2575
@@ -13,24 +13,24 @@ ocnn/models/unet.py,sha256=1FZbTvmWg6sMYkcZyNxcSr_qN6bfOOag-tIwAoqIPKU,4123
 ocnn/modules/__init__.py,sha256=BAEZybvtwDQf7yPZpGcptH1Oxf654Alq87G9D0vKw-E,820
 ocnn/modules/modules.py,sha256=7VlBsbwN49J8Xea0DXcFHVhN8POxhzC3FNMY3_JHukM,6207
 ocnn/modules/resblocks.py,sha256=Xh5EH6aSx8zcLk3D4m4lZiWDejLMu5AQDH7ZDI4p23A,4548
-ocnn/nn/__init__.py,sha256=rmdXQiUa9-GVvVoAxDYf2uTc0yeiwW67BCgOgnqzn60,1707
+ocnn/nn/__init__.py,sha256=6M3GEbyepHCU7l_QU1blSI98M0X4d3jbCsXmVGtXYj0,1769
 ocnn/nn/octree2col.py,sha256=07BGGJD0je0VD-VdS_aKtDo7gWKNWdgojeL1a2n4VRQ,2137
 ocnn/nn/octree2vox.py,sha256=QgyxxZvRvw2taHFWCvcGyCFAnFC6D6nfw71HrvwN3PI,1524
 ocnn/nn/octree_align.py,sha256=Y12GBKS-F3JtRNaDdDcBFDBvhndoRxeMOVua4RB5HZE,1696
-ocnn/nn/octree_conv.py,sha256=rylone_moG8oiWrhb4YYfTgtKcFqHyQ_hCk-VgGzvzI,14468
+ocnn/nn/octree_conv.py,sha256=a6-lCkJR5NJL7lBKZNWklzQUnDLj6hIysGnr45_xCQk,14942
 ocnn/nn/octree_drop.py,sha256=croMHtk0JScDT0nLpdmbiMnkM_b5uVAz6sOEUcta6sY,1963
-ocnn/nn/octree_dwconv.py,sha256=lXwDtcIYUSjuY4qBGIMOsWHYBr5V7L6RFc78B0z8Ki8,6876
+ocnn/nn/octree_dwconv.py,sha256=cIghi7zyMUGhTd6QsU4PS6K2VlwXaYeLNfZFt2xuVi4,7350
 ocnn/nn/octree_gconv.py,sha256=Ogkn7wE49dDdP0X_aBthNfJXMsqcRn3ufJuYQtEEjUI,2928
 ocnn/nn/octree_interp.py,sha256=yQVjiKMNLU1XakPeYjW5ArMsw2fxFRZdh90Nn3cWQuE,7108
 ocnn/nn/octree_norm.py,sha256=Mbn28Hv-CEWt2WA0Pdhj2p127m10vwhepzCJVjiZYMI,2976
 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=w5L3cfXIbyzzQEqC8JiVMX-Jyijh6DxhXulE4voKfm0,23722
-ocnn/octree/points.py,sha256=8yYHVc7E40qA8c3MGuI50k_51nwjg92PXXA3YRXt-LY,11121
+ocnn/octree/octree.py,sha256=PgWiECd3h49AZHrK_0GxwwAgnG9l37AcJILanM7d1-k,23713
+ocnn/octree/points.py,sha256=7_iiN0y0g9SgV8kLzuGW20C-MFOmz5hh1aQhDhUIa0Q,11355
 ocnn/octree/shuffled_key.py,sha256=UJZ4eKNA_7nLbf9FbEvS_3VyrAqnZCOzk1hsPtJianM,3936
-ocnn-2.2.2.dist-info/LICENSE,sha256=YeOS0Plo8Uistv_8ZXdgddmN9GHJKnIiJ5FZ8zTW6Sw,1114
-ocnn-2.2.2.dist-info/METADATA,sha256=tnq-YTGa06xrggKxndRvoe2GS5Hv6qIx65LKyQmF1HM,3297
-ocnn-2.2.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-ocnn-2.2.2.dist-info/top_level.txt,sha256=ayZdVOnxlOke3kgzAlrRh2IEL_qOudwOaEU3xhjtpZ0,5
-ocnn-2.2.2.dist-info/RECORD,,
+ocnn-2.2.3.dist-info/LICENSE,sha256=YeOS0Plo8Uistv_8ZXdgddmN9GHJKnIiJ5FZ8zTW6Sw,1114
+ocnn-2.2.3.dist-info/METADATA,sha256=Uh2UlUwXzRfM9G9d1EBRIiZARcumPCVZPY3CLjR2q8U,3682
+ocnn-2.2.3.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ocnn-2.2.3.dist-info/top_level.txt,sha256=ayZdVOnxlOke3kgzAlrRh2IEL_qOudwOaEU3xhjtpZ0,5
+ocnn-2.2.3.dist-info/RECORD,,

{ocnn-2.2.2.dist-info → ocnn-2.2.3.dist-info}/WHEEL

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