learning3d 0.0.7__py3-none-any.whl → 0.1.0__py3-none-any.whl

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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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/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.7.dist-info → learning3d-0.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: learning3d
-Version: 0.0.7
+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.7.dist-info → learning3d-0.1.0.dist-info}/RECORD

@@ -2,27 +2,27 @@ learning3d/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 learning3d/data_utils/__init__.py,sha256=iYAVh0FThnVlG42QIgmDYrC3NGVYuzKX8s1oRqAI1YU,261
 learning3d/data_utils/dataloaders.py,sha256=QlRd7c7ir19bMliZXq9PF-iNrQ_BtNTonpBCpmjYiFU,14675
 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
-learning3d/examples/test_masknet.py,sha256=dkqUui9sv2SzHtvtzUzL_PxJxMcBMqCSDPAYg0BWAVU,6405
-learning3d/examples/test_masknet2.py,sha256=3_XWBQOwQjK3BCQ__BmPhCvYI_0hZMK3X4C-P2Krw6w,6859
-learning3d/examples/test_pcn.py,sha256=4eaosjJVqiFxlqnaUWu-O2Jawt4uU16UJEzitIjP2us,4342
-learning3d/examples/test_pcrnet.py,sha256=_x9l55sMBACXUfQHLaH7GJMfz6PWdYWSxjrRxVoY-As,4366
-learning3d/examples/test_pnlk.py,sha256=9u7B--PuCpl6TAmObmIKfDvYW6bMj3Jcc3_djTDO-D4,4456
-learning3d/examples/test_pointconv.py,sha256=NUcLjkkNJsGZYaUZHug6QGybm8NshZfg2tc8rfksNU8,4673
-learning3d/examples/test_pointnet.py,sha256=VKfB5DE8fh2G1iIoY02GvjSgWUJk2jQBmmJs3HC5rVU,4324
-learning3d/examples/test_prnet.py,sha256=ls1wWul26UHhquXt-PEn4O2-vCzCdxJ-aNIZDnHfvgg,4901
-learning3d/examples/test_rpmnet.py,sha256=oy-z7I26IQxr4TD_p0qCRnOn6H8VbQFyiWO83ZSFcDk,4476
-learning3d/examples/train_PointNetLK.py,sha256=0GgT2NYKNZl8o02rvW-nYBO_1tlfDNuakuAXtm1V16c,8773
-learning3d/examples/train_dcp.py,sha256=SQVrwnZqGmFCZv_X2tzMysBmv-HI9sllZMWw5zsW3NM,9511
-learning3d/examples/train_deepgmr.py,sha256=vxdkgfQZPtwuYryR0chegTiLuXOQag8r_ccGJ6qtw7o,9397
-learning3d/examples/train_masknet.py,sha256=XzgWsmVAm5Lk21mH9qhvNN0um4pI1fYVfsBAV4deSOM,8889
-learning3d/examples/train_pcn.py,sha256=X7MSYVXwgIMExplua1M9pG20eNhZ_0p83yTADSYrAlA,7542
-learning3d/examples/train_pcrnet.py,sha256=KQ8MiDUiR46qS9t7tc5POJ3NjMyZFBEPOVQY-7Vszpk,8198
-learning3d/examples/train_pointconv.py,sha256=noGT2yGWHAuecObz1X9cEiWl0xjh7NhmRneP88jR8uI,8939
-learning3d/examples/train_pointnet.py,sha256=SXheDRP_GHZQQw4BEYS7bfL481D8GcbTekON-GVwOsk,8840
-learning3d/examples/train_prnet.py,sha256=2zvd-3cYzZP8L92XJmFL5rTzxpApUhetiEc4u4V0X5g,8373
-learning3d/examples/train_rpmnet.py,sha256=PEdFgPXyeME0axvLEd--VbpbqWV6P5i6NnjQnJ_X3Oo,8530
+learning3d/examples/test_dcp.py,sha256=EYXgCz0yh2-bjJE9s8XEUY96LfUE9_hZcdMffyrW3x0,5783
+learning3d/examples/test_deepgmr.py,sha256=f9nv7Y5se9_iZ3UGasTUDnErBZh3FX0fRNY4IqeqjCk,5774
+learning3d/examples/test_masknet.py,sha256=-xiGaaG9SN_u1POh51sM358QFsyx9QoYIhaNWtBNCXY,6558
+learning3d/examples/test_masknet2.py,sha256=ABNDRa9R6Sp9S_H1euj2bx3LnLt6ONKRNvp5bat0jAQ,7012
+learning3d/examples/test_pcn.py,sha256=rvKsqbCRmj4CLU1xWI-VzUh63t89QY6sIHi-39uhzmM,4519
+learning3d/examples/test_pcrnet.py,sha256=JTq7rQN5cFxucVD-0gQ6t0K0ASuX4b_sNI7uxgUKsdg,4519
+learning3d/examples/test_pnlk.py,sha256=WhZfD9s-YUg-nne_JCOSIcs0akJJV30G19QBkMVK2gc,4609
+learning3d/examples/test_pointconv.py,sha256=CA-Z3KLGXtqkCWIUi1IIlpaHCyTMMd5QuNyi37D8Iz8,4826
+learning3d/examples/test_pointnet.py,sha256=Cd0h-x4UcuiV-hDVMeriOYN75-uc_ZQ34mn2ikLEIbg,4477
+learning3d/examples/test_prnet.py,sha256=F5Ju6BafuDocPa3qMa9ZDASaE9zP_5vgp1hgRkpIvso,5078
+learning3d/examples/test_rpmnet.py,sha256=dKORbcOuil2qlUrY5VdHtaUUOGxDc8Paj7FVYmcHqEs,4629
+learning3d/examples/train_PointNetLK.py,sha256=MnYJuU1ThTLRSuylqXZp6Eakeggk2Xd99BJuTyVIvxI,8950
+learning3d/examples/train_dcp.py,sha256=QcyFKKCKwUH3RDUQU7-eET7Q3tIc5hLGlBBrMwv4ans,9688
+learning3d/examples/train_deepgmr.py,sha256=I9GBE78u5tVrKo0zmRDjrTk0Lq-R8TRH27mEKYJU5us,9574
+learning3d/examples/train_masknet.py,sha256=MRlySX4Llfp_0Eu8jwkHg3FdLlWLtlB4tVUNFbSsBbM,9066
+learning3d/examples/train_pcn.py,sha256=QKJLY31Hcx8w2K4XhmY3U3zho5vFePEWr7xHo4fAUvY,7719
+learning3d/examples/train_pcrnet.py,sha256=pUBYK1gtDFhBBon1UBF2yB5-2cGhILkntmplOYRBqiQ,8375
+learning3d/examples/train_pointconv.py,sha256=Vn_WghOFYOOiL-U7yJQ3TB7osYTLbwEQ1WVWTvBhPVs,9116
+learning3d/examples/train_pointnet.py,sha256=bpPMt0QtUoJlVgoMoPX191sO25VCk14pIKdBTCPS62I,9017
+learning3d/examples/train_prnet.py,sha256=aruTiJB8WCad0cbsu5XdfhMSybdibWSGRPilbzi6xWI,8550
+learning3d/examples/train_rpmnet.py,sha256=Q_kB7UnRdcIpFk_w47oNcuhIsmHLLij4dS7vz2GzTIY,8707
 learning3d/losses/__init__.py,sha256=zjjZeA_NvAhZlxiYBbtgjEsvMyLFhFWXlZioitrlGWw,425
 learning3d/losses/chamfer_distance.py,sha256=UTZ6x5cGwL3L5hJZOWoC35gTzcKh1S0yCg8vGuGXU1w,2121
 learning3d/losses/classification.py,sha256=QgDHC5VgSga6BiuD4Ee70t6vvchlE97BY0KExevhdgk,374
@@ -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.7.dist-info/LICENSE,sha256=3qY3_NeQIvalbLlsHFtOfuUKjs_U2k6u7rf6YVx6ac0,1098
-learning3d-0.0.7.dist-info/METADATA,sha256=Hq2QQCZAOuibjXcEwY15hn1QSqj4nkiYmxWujeUk9fE,15813
-learning3d-0.0.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-learning3d-0.0.7.dist-info/top_level.txt,sha256=nTmYW8NhbNV1_15DGNpl_OvvSFtQP98sy3qrrHr0eLo,11
-learning3d-0.0.7.dist-info/RECORD,,
+learning3d-0.1.0.dist-info/LICENSE,sha256=3qY3_NeQIvalbLlsHFtOfuUKjs_U2k6u7rf6YVx6ac0,1098
+learning3d-0.1.0.dist-info/METADATA,sha256=kKzkNAYiiVf_t_gv-_OGGr3C1QZWQ5BCKh5sPZUbLts,15813
+learning3d-0.1.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+learning3d-0.1.0.dist-info/top_level.txt,sha256=nTmYW8NhbNV1_15DGNpl_OvvSFtQP98sy3qrrHr0eLo,11
+learning3d-0.1.0.dist-info/RECORD,,