learning3d 0.0.6__tar.gz → 0.1.0__tar.gz

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

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

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

{learning3d-0.0.6 → learning3d-0.1.0}/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.1.0}/src/learning3d/examples/test_dcp.py

@@ -88,6 +88,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -114,8 +116,8 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 	
-	trainset = RegistrationData('DCP', ModelNet40Data(train=True))
-	testset = RegistrationData('DCP', ModelNet40Data(train=False))
+	trainset = RegistrationData('DCP', ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = RegistrationData('DCP', ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_deepgmr.py

@@ -103,6 +103,8 @@ def options():
 						metavar='K', help='No of nearest neighbors to be estimated.')
 	parser.add_argument('--use_rri', default=True, type=bool,
 						help='Find nearest neighbors to estimate features from PointNet.')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for on training
 	parser.add_argument('-j', '--workers', default=4, type=int,
@@ -121,8 +123,8 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 	
-	trainset = RegistrationData('DeepGMR', ModelNet40Data(train=True))
-	testset = RegistrationData('DeepGMR', ModelNet40Data(train=False))
+	trainset = RegistrationData('DeepGMR', ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = RegistrationData('DeepGMR', ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_masknet.py

@@ -117,6 +117,8 @@ def options():
 						help='Add noise in source point clouds.')
 	parser.add_argument('--outliers', default=False, type=bool,
 						help='Add outliers to template point cloud.')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for on testing
 	parser.add_argument('-j', '--workers', default=1, type=int,
@@ -137,7 +139,7 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points),
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points, root_dir=args.root_dir),
 									partial_source=args.partial_source, noise=args.noise,
 									additional_params={'use_masknet': True})
 	test_loader = DataLoader(testset, batch_size=args.test_batch_size, shuffle=False, drop_last=False, num_workers=args.workers)

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_masknet2.py

@@ -120,6 +120,8 @@ def options():
 						help='Add noise in source point clouds.')
 	parser.add_argument('--outliers', default=False, type=bool,
 						help='Add outliers to template point cloud.')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for on testing
 	parser.add_argument('-j', '--workers', default=1, type=int,
@@ -140,7 +142,7 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points),
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points, root_dir=args.root_dir),
 									partial_template=args.partial_template, partial_source=args.partial_source, 
 									noise=args.noise, additional_params={'use_masknet': True, 'partial_point_cloud_method': 'planar_crop'})
 	test_loader = DataLoader(testset, batch_size=args.test_batch_size, shuffle=False, drop_last=False, num_workers=args.workers)

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_pcn.py

@@ -70,6 +70,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PCN
 	parser.add_argument('--emb_dims', default=1024, type=int,
@@ -95,8 +97,8 @@ def main():
 	args = options()
 	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
 
-	trainset = ClassificationData(ModelNet40Data(train=True))
-	testset = ClassificationData(ModelNet40Data(train=False))
+	trainset = ClassificationData(ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_pcrnet.py

@@ -74,6 +74,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--emb_dims', default=1024, type=int,
@@ -97,7 +99,7 @@ def options():
 def main():
 	args = options()
 
-	testset = RegistrationData('PCRNet', ModelNet40Data(train=False))
+	testset = RegistrationData('PCRNet', ModelNet40Data(train=False, root_dir=args.root_dir))
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 
 	if not torch.cuda.is_available():

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_pnlk.py

@@ -74,6 +74,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--emb_dims', default=1024, type=int,
@@ -98,7 +100,7 @@ def options():
 def main():
 	args = options()
 
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False))
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, root_dir=args.root_dir))
 	test_loader = DataLoader(testset, batch_size=8, shuffle=False, drop_last=False, num_workers=args.workers)
 
 	if not torch.cuda.is_available():

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_pointconv.py

@@ -73,6 +73,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -99,7 +101,7 @@ def main():
 	args = options()
 	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
 	
-	testset = ClassificationData(ModelNet40Data(train=False))
+	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 
 	if not torch.cuda.is_available():

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_pointnet.py

@@ -73,6 +73,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -99,7 +101,7 @@ def main():
 	args = options()
 	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
 	
-	testset = ClassificationData(ModelNet40Data(train=False))
+	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 
 	if not torch.cuda.is_available():

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_prnet.py

@@ -79,6 +79,8 @@ def options():
 	# settings for input data
 	parser.add_argument('--dataset_type', default='modelnet', choices=['modelnet', 'shapenet2'],
 						metavar='DATASET', help='dataset type (default: modelnet)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--emb_dims', default=512, type=int,
@@ -102,8 +104,8 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 	
-	trainset = RegistrationData('PRNet', ModelNet40Data(train=True), partial_source=True, partial_template=True)
-	testset = RegistrationData('PRNet', ModelNet40Data(train=False), partial_source=True, partial_template=True)
+	trainset = RegistrationData('PRNet', ModelNet40Data(train=True, root_dir=args.root_dir), partial_source=True, partial_template=True)
+	testset = RegistrationData('PRNet', ModelNet40Data(train=False, root_dir=args.root_dir), partial_source=True, partial_template=True)
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/test_rpmnet.py

@@ -74,6 +74,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--emb_dims', default=1024, type=int,
@@ -98,7 +100,7 @@ def options():
 def main():
 	args = options()
 
-	testset = RegistrationData('RPMNet', ModelNet40Data(train=False, num_points=args.num_points, use_normals=True), partial_source=True, partial_template=False)
+	testset = RegistrationData('RPMNet', ModelNet40Data(train=False, num_points=args.num_points, use_normals=True, root_dir=args.root_dir), partial_source=True, partial_template=False)
 	test_loader = DataLoader(testset, batch_size=1, shuffle=False, drop_last=False, num_workers=args.workers)
 
 	if not torch.cuda.is_available():

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_PointNetLK.py

@@ -147,6 +147,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--fine_tune_pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -195,8 +197,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = RegistrationData('PointNetLK', ModelNet40Data(train=True))
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False))
+	trainset = RegistrationData('PointNetLK', ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=8, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_dcp.py

@@ -168,6 +168,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -214,8 +216,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = RegistrationData('DCP', ModelNet40Data(train=True))
-	testset = RegistrationData('DCP', ModelNet40Data(train=False))
+	trainset = RegistrationData('DCP', ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = RegistrationData('DCP', ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_deepgmr.py

@@ -165,6 +165,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	parser.add_argument('--nearest_neighbors', default=20, type=int,
 						metavar='K', help='No of nearest neighbors to be estimated.')
@@ -211,8 +213,8 @@ def main():
 	textio = IOStream('checkpoints/' + args.exp_name + '/run.log')
 	textio.cprint(str(args))
 	
-	trainset = RegistrationData('DeepGMR', ModelNet40Data(train=True), additional_params={'nearest_neighbors': args.nearest_neighbors})
-	testset = RegistrationData('DeepGMR', ModelNet40Data(train=False), additional_params={'nearest_neighbors': args.nearest_neighbors})
+	trainset = RegistrationData('DeepGMR', ModelNet40Data(train=True, root_dir=args.root_dir), additional_params={'nearest_neighbors': args.nearest_neighbors})
+	testset = RegistrationData('DeepGMR', ModelNet40Data(train=False, root_dir=args.root_dir), additional_params={'nearest_neighbors': args.nearest_neighbors})
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_masknet.py

@@ -160,6 +160,8 @@ def options():
 						help='Add noise in source point clouds.')
 	parser.add_argument('--outliers', default=False, type=bool,
 						help='Add outliers to template point cloud.')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for on training
 	parser.add_argument('--seed', type=int, default=1234)
@@ -202,10 +204,10 @@ def main():
 	textio = IOStream('checkpoints/' + args.exp_name + '/run.log')
 	textio.cprint(str(args))
 
-	trainset = RegistrationData(ModelNet40Data(train=True, num_points=args.num_points, unseen=args.unseen),
+	trainset = RegistrationData(ModelNet40Data(train=True, num_points=args.num_points, unseen=args.unseen, root_dir=args.root_dir),
 								partial_source=args.partial_source, noise=args.noise, outliers=args.outliers, 
 								additional_params={'use_masknet': True})
-	testset = RegistrationData(ModelNet40Data(train=False, num_points=args.num_points, unseen=args.unseen),
+	testset = RegistrationData(ModelNet40Data(train=False, num_points=args.num_points, unseen=args.unseen, root_dir=args.root_dir),
 								partial_source=args.partial_source, noise=args.noise, outliers=args.outliers,
 								additional_params={'use_masknet': True})
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_pcn.py

@@ -138,6 +138,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PCN
 	parser.add_argument('--emb_dims', default=1024, type=int,
@@ -183,8 +185,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = ClassificationData(ModelNet40Data(train=True))
-	testset = ClassificationData(ModelNet40Data(train=False))
+	trainset = ClassificationData(ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_pcrnet.py

@@ -147,6 +147,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -193,8 +195,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = RegistrationData('PCRNet', ModelNet40Data(train=True))
-	testset = RegistrationData('PCRNet', ModelNet40Data(train=False))
+	trainset = RegistrationData('PCRNet', ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = RegistrationData('PCRNet', ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_pointconv.py

@@ -163,6 +163,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -210,8 +212,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = ClassificationData(ModelNet40Data(train=True))
-	testset = ClassificationData(ModelNet40Data(train=False))
+	trainset = ClassificationData(ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_pointnet.py

@@ -163,6 +163,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -210,8 +212,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = ClassificationData(ModelNet40Data(train=True))
-	testset = ClassificationData(ModelNet40Data(train=False))
+	trainset = ClassificationData(ModelNet40Data(train=True, root_dir=args.root_dir))
+	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_prnet.py

@@ -157,6 +157,8 @@ def options():
 						metavar='K', help='dim. of the feature vector (default: 1024)')
 	parser.add_argument('--num_iterations', default=3, type=int,
 						help='Number of Iterations')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for on training
 	parser.add_argument('--seed', type=int, default=1234)
@@ -195,8 +197,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = RegistrationData('PRNet', ModelNet40Data(train=True), partial_source=True, partial_template=True)
-	testset = RegistrationData('PRNet', ModelNet40Data(train=False), partial_source=True, partial_template=True)
+	trainset = RegistrationData('PRNet', ModelNet40Data(train=True, root_dir=args.root_dir), partial_source=True, partial_template=True)
+	testset = RegistrationData('PRNet', ModelNet40Data(train=False, root_dir=args.root_dir), partial_source=True, partial_template=True)
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
 

{learning3d-0.0.6 → learning3d-0.1.0}/src/learning3d/examples/train_rpmnet.py

@@ -146,6 +146,8 @@ def options():
 						metavar='DATASET', help='dataset type (default: modelnet)')
 	parser.add_argument('--num_points', default=1024, type=int,
 						metavar='N', help='points in point-cloud (default: 1024)')
+	parser.add_argument('--root_dir', default='./', type=str, 
+					 	help='path of the data where modelnet files are downloaded.')
 
 	# settings for PointNet
 	parser.add_argument('--fine_tune_pointnet', default='tune', type=str, choices=['fixed', 'tune'],
@@ -194,8 +196,8 @@ def main():
 	textio.cprint(str(args))
 
 	
-	trainset = RegistrationData('RPMNet', ModelNet40Data(train=True, num_points=args.num_points, use_normals=True), partial_source=True, partial_template=True)
-	testset = RegistrationData('RPMNet', ModelNet40Data(train=False, num_points=args.num_points, use_normals=True), partial_source=True, partial_template=True)
+	trainset = RegistrationData('RPMNet', ModelNet40Data(train=True, num_points=args.num_points, use_normals=True, root_dir=args.root_dir), partial_source=True, partial_template=True)
+	testset = RegistrationData('RPMNet', ModelNet40Data(train=False, num_points=args.num_points, use_normals=True, root_dir=args.root_dir), partial_source=True, partial_template=True)
 	train_loader = DataLoader(trainset, batch_size=args.batch_size, shuffle=True, drop_last=True, num_workers=args.workers)
 	test_loader = DataLoader(testset, batch_size=8, shuffle=False, drop_last=False, num_workers=args.workers)
 

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

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