learning3d 0.0.5__tar.gz → 0.0.7__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (90) hide show
  1. {learning3d-0.0.5/src/learning3d.egg-info → learning3d-0.0.7}/PKG-INFO +1 -1
  2. {learning3d-0.0.5 → learning3d-0.0.7}/pyproject.toml +1 -1
  3. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/data_utils/dataloaders.py +4 -3
  4. {learning3d-0.0.5 → learning3d-0.0.7/src/learning3d.egg-info}/PKG-INFO +1 -1
  5. {learning3d-0.0.5 → learning3d-0.0.7}/LICENSE +0 -0
  6. {learning3d-0.0.5 → learning3d-0.0.7}/MANIFEST.in +0 -0
  7. {learning3d-0.0.5 → learning3d-0.0.7}/README.md +0 -0
  8. {learning3d-0.0.5 → learning3d-0.0.7}/data/modelnet40_ply_hdf5_2048/shape_names.txt +0 -0
  9. {learning3d-0.0.5 → learning3d-0.0.7}/data/modelnet40_ply_hdf5_2048/test_files.txt +0 -0
  10. {learning3d-0.0.5 → learning3d-0.0.7}/data/modelnet40_ply_hdf5_2048/train_files.txt +0 -0
  11. {learning3d-0.0.5 → learning3d-0.0.7}/requirements.txt +0 -0
  12. {learning3d-0.0.5 → learning3d-0.0.7}/setup.cfg +0 -0
  13. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/__init__.py +0 -0
  14. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/data_utils/__init__.py +0 -0
  15. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/data_utils/user_data.py +0 -0
  16. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_dcp.py +0 -0
  17. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_deepgmr.py +0 -0
  18. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_masknet.py +0 -0
  19. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_masknet2.py +0 -0
  20. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_pcn.py +0 -0
  21. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_pcrnet.py +0 -0
  22. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_pnlk.py +0 -0
  23. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_pointconv.py +0 -0
  24. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_pointnet.py +0 -0
  25. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_prnet.py +0 -0
  26. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/test_rpmnet.py +0 -0
  27. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_PointNetLK.py +0 -0
  28. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_dcp.py +0 -0
  29. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_deepgmr.py +0 -0
  30. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_masknet.py +0 -0
  31. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_pcn.py +0 -0
  32. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_pcrnet.py +0 -0
  33. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_pointconv.py +0 -0
  34. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_pointnet.py +0 -0
  35. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_prnet.py +0 -0
  36. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/examples/train_rpmnet.py +0 -0
  37. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/__init__.py +0 -0
  38. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/chamfer_distance.py +0 -0
  39. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/classification.py +0 -0
  40. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/correspondence_loss.py +0 -0
  41. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/chamfer_distance/__init__.py +0 -0
  42. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/chamfer_distance/chamfer_distance.cpp +0 -0
  43. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/chamfer_distance/chamfer_distance.cu +0 -0
  44. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/chamfer_distance/chamfer_distance.py +0 -0
  45. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/emd_loss_layer.py +0 -0
  46. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/include/cuda/emd.cuh +0 -0
  47. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/include/cuda_helper.h +0 -0
  48. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/include/emd.h +0 -0
  49. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/layer/__init__.py +0 -0
  50. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/layer/emd_loss_layer.py +0 -0
  51. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/src/cuda/emd.cu +0 -0
  52. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/pkg/src/emd.cpp +0 -0
  53. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/cuda/emd_torch/setup.py +0 -0
  54. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/emd.py +0 -0
  55. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/frobenius_norm.py +0 -0
  56. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/losses/rmse_features.py +0 -0
  57. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/__init__.py +0 -0
  58. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/classifier.py +0 -0
  59. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/dcp.py +0 -0
  60. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/deepgmr.py +0 -0
  61. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/dgcnn.py +0 -0
  62. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/masknet.py +0 -0
  63. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/masknet2.py +0 -0
  64. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/pcn.py +0 -0
  65. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/pcrnet.py +0 -0
  66. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/pointconv.py +0 -0
  67. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/pointnet.py +0 -0
  68. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/pointnetlk.py +0 -0
  69. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/pooling.py +0 -0
  70. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/ppfnet.py +0 -0
  71. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/prnet.py +0 -0
  72. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/rpmnet.py +0 -0
  73. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/models/segmentation.py +0 -0
  74. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/__init__.py +0 -0
  75. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/data_utils.py +0 -0
  76. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/invmat.py +0 -0
  77. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/quaternion.py +0 -0
  78. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/se3.py +0 -0
  79. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/sinc.py +0 -0
  80. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/so3.py +0 -0
  81. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/ops/transform_functions.py +0 -0
  82. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/utils/__init__.py +0 -0
  83. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/utils/pointconv_util.py +0 -0
  84. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/utils/ppfnet_util.py +0 -0
  85. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/utils/svd.py +0 -0
  86. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/utils/transformer.py +0 -0
  87. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d.egg-info/SOURCES.txt +0 -0
  88. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d.egg-info/dependency_links.txt +0 -0
  89. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d.egg-info/requires.txt +0 -0
  90. {learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d.egg-info/top_level.txt +0 -0
{learning3d-0.0.5/src/learning3d.egg-info → learning3d-0.0.7}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: learning3d
3
- Version: 0.0.5
3
+ Version: 0.0.7
4
4
  Summary: Learning3D: A Modern Library for Deep Learning on 3D Point Clouds Data
5
5
  Author-email: Vinit Sarode <vinitsarode5@gmail.com>
6
6
  Project-URL: Homepage, https://github.com/vinits5/learning3d
{learning3d-0.0.5 → learning3d-0.0.7}/pyproject.toml RENAMED
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "learning3d"
7
- version = "0.0.5"
7
+ version = "0.0.7"
8
8
  authors = [
9
9
  { name="Vinit Sarode", email="vinitsarode5@gmail.com"},
10
10
  ]
{learning3d-0.0.5 → learning3d-0.0.7}/src/learning3d/data_utils/dataloaders.py RENAMED
@@ -184,13 +184,14 @@ class ModelNet40Data(Dataset):
184
184
  self,
185
185
  train=True,
186
186
  num_points=1024,
187
+ download=False,
187
188
  root_dir='./',
188
189
  randomize_data=False,
189
190
  use_normals=False
190
191
  ):
191
192
  super(ModelNet40Data, self).__init__()
192
193
  self.root_dir = root_dir
193
- if root_dir is not None: download_modelnet40(root_dir=root_dir)
194
+ if download: download_modelnet40(root_dir=root_dir)
194
195
  self.data, self.labels = load_data(root_dir, train, use_normals)
195
196
  if not train: self.shapes = self.read_classes_ModelNet40()
196
197
  self.num_points = num_points
@@ -225,7 +226,7 @@ class ModelNet40Data(Dataset):
225
226
 
226
227
 
227
228
  class ClassificationData(Dataset):
228
- def __init__(self, data_class=ModelNet40Data()):
229
+ def __init__(self, data_class):
229
230
  super(ClassificationData, self).__init__()
230
231
  self.set_class(data_class)
231
232
 
@@ -246,7 +247,7 @@ class ClassificationData(Dataset):
246
247
 
247
248
 
248
249
  class RegistrationData(Dataset):
249
- def __init__(self, algorithm, data_class=ModelNet40Data(), partial_source=False, partial_template=False, noise=False, additional_params={}):
250
+ def __init__(self, algorithm, data_class, partial_source=False, partial_template=False, noise=False, additional_params={}):
250
251
  super(RegistrationData, self).__init__()
251
252
  available_algorithms = ['PCRNet', 'PointNetLK', 'DCP', 'PRNet', 'iPCRNet', 'RPMNet', 'DeepGMR']
252
253
  if algorithm in available_algorithms: self.algorithm = algorithm
{learning3d-0.0.5 → learning3d-0.0.7/src/learning3d.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: learning3d
3
- Version: 0.0.5
3
+ Version: 0.0.7
4
4
  Summary: Learning3D: A Modern Library for Deep Learning on 3D Point Clouds Data
5
5
  Author-email: Vinit Sarode <vinitsarode5@gmail.com>
6
6
  Project-URL: Homepage, https://github.com/vinits5/learning3d
{learning3d-0.0.5 → learning3d-0.0.7}/LICENSE RENAMED
File without changes
{learning3d-0.0.5 → learning3d-0.0.7}/MANIFEST.in RENAMED
File without changes
{learning3d-0.0.5 → learning3d-0.0.7}/README.md RENAMED
File without changes
{learning3d-0.0.5 → learning3d-0.0.7}/requirements.txt RENAMED
File without changes
{learning3d-0.0.5 → learning3d-0.0.7}/setup.cfg RENAMED
File without changes