learning3d 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

learning3d/data_utils/dataloaders.py

@@ -184,13 +184,14 @@ class ModelNet40Data(Dataset):
 		self,
 		train=True,
 		num_points=1024,
-		root_dir=None,
+		download=False,
+		root_dir='./',
 		randomize_data=False,
 		use_normals=False
 	):
 		super(ModelNet40Data, self).__init__()
 		self.root_dir = root_dir
-		if root_dir is not None: download_modelnet40(root_dir=root_dir)
+		if download: download_modelnet40(root_dir=root_dir)
 		self.data, self.labels = load_data(root_dir, train, use_normals)
 		if not train: self.shapes = self.read_classes_ModelNet40()
 		self.num_points = num_points
@@ -225,15 +226,15 @@ class ModelNet40Data(Dataset):
 
 
 class ClassificationData(Dataset):
-	def __init__(self, data_class=ModelNet40Data()):
+	def __init__(self, data_class=ModelNet40Data, root_dir='./'):
 		super(ClassificationData, self).__init__()
-		self.set_class(data_class)
+		self.set_class(data_class, root_dir)
 
 	def __len__(self):
 		return len(self.data_class)
 
-	def set_class(self, data_class):
-		self.data_class = data_class
+	def set_class(self, data_class, root_dir):
+		self.data_class = data_class(root_dir=root_dir)
 
 	def get_shape(self, label):
 		try:
@@ -246,13 +247,13 @@ class ClassificationData(Dataset):
 
 
 class RegistrationData(Dataset):
-	def __init__(self, algorithm, data_class=ModelNet40Data(), partial_source=False, partial_template=False, noise=False, additional_params={}):
+	def __init__(self, algorithm, root_dir='./', data_class=ModelNet40Data, partial_source=False, partial_template=False, noise=False, additional_params={}):
 		super(RegistrationData, self).__init__()
 		available_algorithms = ['PCRNet', 'PointNetLK', 'DCP', 'PRNet', 'iPCRNet', 'RPMNet', 'DeepGMR']
 		if algorithm in available_algorithms: self.algorithm = algorithm
 		else: raise Exception("Algorithm not available for registration.")
 		
-		self.set_class(data_class)
+		self.set_class(data_class, root_dir)
 		self.partial_template = partial_template
 		self.partial_source = partial_source
 		self.noise = noise
@@ -282,8 +283,8 @@ class RegistrationData(Dataset):
 	def __len__(self):
 		return len(self.data_class)
 
-	def set_class(self, data_class):
-		self.data_class = data_class
+	def set_class(self, data_class, root_dir):
+		self.data_class = data_class(root_dir=root_dir)
 
 	def __getitem__(self, index):
 		template, label = self.data_class[index]

{learning3d-0.0.4.dist-info → learning3d-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: learning3d
-Version: 0.0.4
+Version: 0.0.6
 Summary: Learning3D: A Modern Library for Deep Learning on 3D Point Clouds Data
 Author-email: Vinit Sarode <vinitsarode5@gmail.com>
 Project-URL: Homepage, https://github.com/vinits5/learning3d

{learning3d-0.0.4.dist-info → learning3d-0.0.6.dist-info}/RECORD

@@ -1,6 +1,6 @@
 learning3d/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 learning3d/data_utils/__init__.py,sha256=iYAVh0FThnVlG42QIgmDYrC3NGVYuzKX8s1oRqAI1YU,261
-learning3d/data_utils/dataloaders.py,sha256=XqlVl-IumBxW1woA8GtgCgEMNhrjCgyaAtKhYxPxnTg,14703
+learning3d/data_utils/dataloaders.py,sha256=s4eNc3ISqSutScvXkVZ-6PMY0-Bu5Br_N7XUhPAFjHs,14813
 learning3d/data_utils/user_data.py,sha256=ADDGeCUCr6TcXhcxvAFncIeLO71xoRHYi4H418ktvQs,4828
 learning3d/examples/test_dcp.py,sha256=o1hgl22b2xhqqYe4_k7NlNo3-rIMmSzzssCI74DCfoE,5606
 learning3d/examples/test_deepgmr.py,sha256=pb-gRE6YLLaPFmiBXQjaE-B-J314QY4RkNwLHx65bcU,5597
@@ -73,8 +73,8 @@ learning3d/utils/pointconv_util.py,sha256=kJxGztai7X15YsGuorMOc50SPtj_k1yfkP4XCT
 learning3d/utils/ppfnet_util.py,sha256=HEoxkgUBlawKZLWspfQm3caWUyAMIrW-ECtStNYbe2Y,7989
 learning3d/utils/svd.py,sha256=yCYQt2SKqeIzCBnBEr_8xFR79m4fIoNVFnp77epn1dM,1936
 learning3d/utils/transformer.py,sha256=UDgJvnh7ekWyijaAn-a3ckeFeMxlK_chXzWlhAGDiPM,8974
-learning3d-0.0.4.dist-info/LICENSE,sha256=3qY3_NeQIvalbLlsHFtOfuUKjs_U2k6u7rf6YVx6ac0,1098
-learning3d-0.0.4.dist-info/METADATA,sha256=zC24gcn1X2bakXZ3q0phrhkXyJXQhkyOWbDA6KncVJk,15813
-learning3d-0.0.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-learning3d-0.0.4.dist-info/top_level.txt,sha256=nTmYW8NhbNV1_15DGNpl_OvvSFtQP98sy3qrrHr0eLo,11
-learning3d-0.0.4.dist-info/RECORD,,
+learning3d-0.0.6.dist-info/LICENSE,sha256=3qY3_NeQIvalbLlsHFtOfuUKjs_U2k6u7rf6YVx6ac0,1098
+learning3d-0.0.6.dist-info/METADATA,sha256=S8XK9N6x5U7C6FqGVQwvAhOkegpGd0M1qLlvD2GvAwI,15813
+learning3d-0.0.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+learning3d-0.0.6.dist-info/top_level.txt,sha256=nTmYW8NhbNV1_15DGNpl_OvvSFtQP98sy3qrrHr0eLo,11
+learning3d-0.0.6.dist-info/RECORD,,