learning3d 0.0.6__tar.gz → 0.0.7__tar.gz

{learning3d-0.0.6/src/learning3d.egg-info → learning3d-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: learning3d
-Version: 0.0.6
+Version: 0.0.7
 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.6 → learning3d-0.0.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "learning3d"
-version = "0.0.6"
+version = "0.0.7"
 authors = [
     { name="Vinit Sarode", email="vinitsarode5@gmail.com"},
 ]

{learning3d-0.0.6 → learning3d-0.0.7}/src/learning3d/data_utils/dataloaders.py

@@ -226,15 +226,15 @@ class ModelNet40Data(Dataset):
 
 
 class ClassificationData(Dataset):
-	def __init__(self, data_class=ModelNet40Data, root_dir='./'):
+	def __init__(self, data_class):
 		super(ClassificationData, self).__init__()
-		self.set_class(data_class, root_dir)
+		self.set_class(data_class)
 
 	def __len__(self):
 		return len(self.data_class)
 
-	def set_class(self, data_class, root_dir):
-		self.data_class = data_class(root_dir=root_dir)
+	def set_class(self, data_class):
+		self.data_class = data_class
 
 	def get_shape(self, label):
 		try:
@@ -247,13 +247,13 @@ class ClassificationData(Dataset):
 
 
 class RegistrationData(Dataset):
-	def __init__(self, algorithm, root_dir='./', data_class=ModelNet40Data, partial_source=False, partial_template=False, noise=False, additional_params={}):
+	def __init__(self, algorithm, data_class, 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, root_dir)
+		self.set_class(data_class)
 		self.partial_template = partial_template
 		self.partial_source = partial_source
 		self.noise = noise
@@ -283,8 +283,8 @@ class RegistrationData(Dataset):
 	def __len__(self):
 		return len(self.data_class)
 
-	def set_class(self, data_class, root_dir):
-		self.data_class = data_class(root_dir=root_dir)
+	def set_class(self, data_class):
+		self.data_class = data_class
 
 	def __getitem__(self, index):
 		template, label = self.data_class[index]

{learning3d-0.0.6 → learning3d-0.0.7/src/learning3d.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: learning3d
-Version: 0.0.6
+Version: 0.0.7
 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