learning3d 0.1.0__tar.gz → 0.2.1__tar.gz

learning3d-0.2.1/MANIFEST.in

@@ -0,0 +1,4 @@
+global-exclude *.cu *.h *.cpp *.egg-info *.so *.o *.egg *.cuh *.txt
+global-exclude *__pycache__* *.pyc *pretrained* *lib/* *cuda/chamfer_distance/* 
+global-exclude *cuda/emd_torch/*/*/* *cuda/emd_torch/*/* *cuda/emd_torch/*
+global-exclude *flownet*

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

@@ -1,10 +1,35 @@
 Metadata-Version: 2.1
 Name: learning3d
-Version: 0.1.0
+Version: 0.2.1
 Summary: Learning3D: A Modern Library for Deep Learning on 3D Point Clouds Data
 Author-email: Vinit Sarode <vinitsarode5@gmail.com>
+Maintainer-email: Vinit Sarode <vinitsarode5@gmail.com>
+License: The MIT License
+        
+        Copyright (c) 2010-2019 Google, Inc. http://angularjs.org
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in
+        all copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+        THE SOFTWARE.
 Project-URL: Homepage, https://github.com/vinits5/learning3d
+Project-URL: Repository, https://github.com/vinits5/learning3d
 Project-URL: Issues, https://github.com/vinits5/learning3d/issues
+Project-URL: Changelog, https://github.com/vinits5/learning3d/CHANGELOG.md
+Keywords: Point Clouds,Deep Learning,3D Vision,Point Cloud Registration,Point Cloud Classification,Point Cloud Segmentation
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
@@ -22,9 +47,10 @@ Requires-Dist: scikit-learn==1.2.2
 Requires-Dist: scipy==1.10.1
 Requires-Dist: numpy==1.24.3
 Requires-Dist: transforms3d==0.4.1
+Requires-Dist: pycuda
 
 <p align="center">
-	<img src="https://github.com/vinits5/learning3d/blob/master/images/logo.png" height="170">
+	<img src="https://github.com/vinits5/learning3d/blob/pypi_v0.2.0/learning3d/images/logo.png" height="170">
 </p>
 
 # Learning3D: A Modern Library for Deep Learning on 3D Point Clouds Data.
@@ -34,24 +60,40 @@ Requires-Dist: transforms3d==0.4.1
 Learning3D is an open-source library that supports the development of deep learning algorithms that deal with 3D data. The Learning3D exposes a set of state of art deep neural networks in python. A modular code has been provided for further development. We welcome contributions from the open-source community.
 
 ## Latest News:
-1. \[24 Oct, 2023\]: [MaskNet++](https://github.com/zhouruqin/MaskNet2) is now a part of learning3d library.
-2. \[12 May, 2022\]: [ChamferDistance](https://github.com/fwilliams/fml) loss function is incorporated in learning3d. This is a purely pytorch based loss function.
-3. \[24 Dec. 2020\]: [MaskNet](https://arxiv.org/pdf/2010.09185.pdf) is now ready to enhance the performance of registration algorithms in learning3d for occluded point clouds.
-4. \[24 Dec. 2020\]: Loss based on the predicted and ground truth correspondences is added in learning3d after consideration of [Correspondence Matrices are Underrated](https://arxiv.org/pdf/2010.16085.pdf) paper.
-5. \[24 Dec. 2020\]: [PointConv](https://arxiv.org/abs/1811.07246), latent feature estimation using convolutions on point clouds is now available in learning3d.
-6. \[16 Oct. 2020\]: [DeepGMR](https://wentaoyuan.github.io/deepgmr/), registration using gaussian mixture models is now available in learning3d
-7. \[14 Oct. 2020\]: Now, use your own data in learning3d. (Check out [UserData](https://github.com/vinits5/learning3d#use-your-own-data) functionality!)
+1. \[28 Feb, 2025\]: [CurveNet](https://github.com/tiangexiang/CurveNet) is now a part of learning3d library.
+2. \[7 Apr, 2024\]: Now, learning3d is available as pypi package.
+3. \[24 Oct, 2023\]: [MaskNet++](https://github.com/zhouruqin/MaskNet2) is now a part of learning3d library.
+4. \[12 May, 2022\]: [ChamferDistance](https://github.com/fwilliams/fml) loss function is incorporated in learning3d. This is a purely pytorch based loss function.
+5. \[24 Dec. 2020\]: [MaskNet](https://arxiv.org/pdf/2010.09185.pdf) is now ready to enhance the performance of registration algorithms in learning3d for occluded point clouds.
+6. \[24 Dec. 2020\]: Loss based on the predicted and ground truth correspondences is added in learning3d after consideration of [Correspondence Matrices are Underrated](https://arxiv.org/pdf/2010.16085.pdf) paper.
+7. \[24 Dec. 2020\]: [PointConv](https://arxiv.org/abs/1811.07246), latent feature estimation using convolutions on point clouds is now available in learning3d.
+8. \[16 Oct. 2020\]: [DeepGMR](https://wentaoyuan.github.io/deepgmr/), registration using gaussian mixture models is now available in learning3d
+9. \[14 Oct. 2020\]: Now, use your own data in learning3d. (Check out [UserData](https://github.com/vinits5/learning3d#use-your-own-data) functionality!)
+
+## PyPI package setup
+### Setup from pypi server
+```
+pip install learning3d
+```
+
+### Setup using code
+```
+git clone https://github.com/vinits5/learning3d.git
+cd learning3d
+git checkout pypi_v0.1.0
+python3 -m pip install .
+```
 
 ## Available Computer Vision Algorithms in Learning3D
 
 | Sr. No.       | Tasks         | Algorithms  |
 |:-------------:|:----------:|:-----|
-| 1 | [Classification](https://github.com/vinits5/learning3d#use-of-classification--segmentation-network) | PointNet, DGCNN, PPFNet, [PointConv](https://github.com/vinits5/learning3d#use-of-pointconv) |
+| 1 | [Classification](https://github.com/vinits5/learning3d#use-of-classification--segmentation-network) | PointNet, DGCNN, PPFNet, [PointConv](https://github.com/vinits5/learning3d#use-of-pointconv), [CurveNet](https://github.com/tiangexiang/CurveNet) |
 | 2 | [Segmentation](https://github.com/vinits5/learning3d#use-of-classification--segmentation-network) | PointNet, DGCNN |
 | 3 | [Reconstruction](https://github.com/vinits5/learning3d#use-of-point-completion-network) | Point Completion Network (PCN) |
 | 4 | [Registration](https://github.com/vinits5/learning3d#use-of-registration-networks) | PointNetLK, PCRNet, DCP, PRNet, RPM-Net, DeepGMR |
 | 5 | [Flow Estimation](https://github.com/vinits5/learning3d#use-of-flow-estimation-network) | FlowNet3D |
-| 6 | [Inlier Estimation](https://github.com/vinits5/learning3d#use-of-inlier-estimation-network-masknet) | MaskNet, MaskNet++ | 
+| 6 | [Inlier Estimation](https://github.com/vinits5/learning3d#use-of-inlier-estimation-network-masknet) | MaskNet, [MaskNet++](https://github.com/zhouruqin/MaskNet2) | 
 
 ## Available Pretrained Models
 1. PointNet
@@ -66,6 +108,7 @@ Learning3D is an open-source library that supports the development of deep learn
 10. PointConv (Download from this [link](https://github.com/DylanWusee/pointconv_pytorch/blob/master/checkpoints/checkpoint.pth))
 11. MaskNet
 12. MaskNet++ / MaskNet2
+13. CurveNet
 
 ## Available Datasets
 1. ModelNet40
@@ -81,7 +124,8 @@ Learning3D is an open-source library that supports the development of deep learn
 1. Ubuntu 16.04
 2. Ubuntu 18.04
 3. Ubuntu 20.04.6
-3. Linux Mint
+4. Linux Mint
+5. macOS Sequoia 15.3.1
 
 ### Requirements
 1. CUDA 10.0 or higher
@@ -268,3 +312,4 @@ PointConv variable is a class. Users can use it to create a sub-class to overrid
 14. [CMU:](https://arxiv.org/pdf/2010.16085.pdf) Correspondence Matrices are Underrated
 15. [MaskNet:](https://arxiv.org/pdf/2010.09185.pdf) A Fully-Convolutional Network to Estimate Inlier Points
 16. [MaskNet++:](https://www.sciencedirect.com/science/article/abs/pii/S0097849322000085) Inlier/outlier identification for two point clouds
+17. [CurveNet:](https://github.com/tiangexiang/CurveNet) Walk in the Cloud: Learning Curves for Point Clouds Shape Analysis

{learning3d-0.1.0 → learning3d-0.2.1}/README.md

@@ -1,5 +1,5 @@
 <p align="center">
-	<img src="https://github.com/vinits5/learning3d/blob/master/images/logo.png" height="170">
+	<img src="https://github.com/vinits5/learning3d/blob/pypi_v0.2.0/learning3d/images/logo.png" height="170">
 </p>
 
 # Learning3D: A Modern Library for Deep Learning on 3D Point Clouds Data.
@@ -9,24 +9,40 @@
 Learning3D is an open-source library that supports the development of deep learning algorithms that deal with 3D data. The Learning3D exposes a set of state of art deep neural networks in python. A modular code has been provided for further development. We welcome contributions from the open-source community.
 
 ## Latest News:
-1. \[24 Oct, 2023\]: [MaskNet++](https://github.com/zhouruqin/MaskNet2) is now a part of learning3d library.
-2. \[12 May, 2022\]: [ChamferDistance](https://github.com/fwilliams/fml) loss function is incorporated in learning3d. This is a purely pytorch based loss function.
-3. \[24 Dec. 2020\]: [MaskNet](https://arxiv.org/pdf/2010.09185.pdf) is now ready to enhance the performance of registration algorithms in learning3d for occluded point clouds.
-4. \[24 Dec. 2020\]: Loss based on the predicted and ground truth correspondences is added in learning3d after consideration of [Correspondence Matrices are Underrated](https://arxiv.org/pdf/2010.16085.pdf) paper.
-5. \[24 Dec. 2020\]: [PointConv](https://arxiv.org/abs/1811.07246), latent feature estimation using convolutions on point clouds is now available in learning3d.
-6. \[16 Oct. 2020\]: [DeepGMR](https://wentaoyuan.github.io/deepgmr/), registration using gaussian mixture models is now available in learning3d
-7. \[14 Oct. 2020\]: Now, use your own data in learning3d. (Check out [UserData](https://github.com/vinits5/learning3d#use-your-own-data) functionality!)
+1. \[28 Feb, 2025\]: [CurveNet](https://github.com/tiangexiang/CurveNet) is now a part of learning3d library.
+2. \[7 Apr, 2024\]: Now, learning3d is available as pypi package.
+3. \[24 Oct, 2023\]: [MaskNet++](https://github.com/zhouruqin/MaskNet2) is now a part of learning3d library.
+4. \[12 May, 2022\]: [ChamferDistance](https://github.com/fwilliams/fml) loss function is incorporated in learning3d. This is a purely pytorch based loss function.
+5. \[24 Dec. 2020\]: [MaskNet](https://arxiv.org/pdf/2010.09185.pdf) is now ready to enhance the performance of registration algorithms in learning3d for occluded point clouds.
+6. \[24 Dec. 2020\]: Loss based on the predicted and ground truth correspondences is added in learning3d after consideration of [Correspondence Matrices are Underrated](https://arxiv.org/pdf/2010.16085.pdf) paper.
+7. \[24 Dec. 2020\]: [PointConv](https://arxiv.org/abs/1811.07246), latent feature estimation using convolutions on point clouds is now available in learning3d.
+8. \[16 Oct. 2020\]: [DeepGMR](https://wentaoyuan.github.io/deepgmr/), registration using gaussian mixture models is now available in learning3d
+9. \[14 Oct. 2020\]: Now, use your own data in learning3d. (Check out [UserData](https://github.com/vinits5/learning3d#use-your-own-data) functionality!)
+
+## PyPI package setup
+### Setup from pypi server
+```
+pip install learning3d
+```
+
+### Setup using code
+```
+git clone https://github.com/vinits5/learning3d.git
+cd learning3d
+git checkout pypi_v0.1.0
+python3 -m pip install .
+```
 
 ## Available Computer Vision Algorithms in Learning3D
 
 | Sr. No.       | Tasks         | Algorithms  |
 |:-------------:|:----------:|:-----|
-| 1 | [Classification](https://github.com/vinits5/learning3d#use-of-classification--segmentation-network) | PointNet, DGCNN, PPFNet, [PointConv](https://github.com/vinits5/learning3d#use-of-pointconv) |
+| 1 | [Classification](https://github.com/vinits5/learning3d#use-of-classification--segmentation-network) | PointNet, DGCNN, PPFNet, [PointConv](https://github.com/vinits5/learning3d#use-of-pointconv), [CurveNet](https://github.com/tiangexiang/CurveNet) |
 | 2 | [Segmentation](https://github.com/vinits5/learning3d#use-of-classification--segmentation-network) | PointNet, DGCNN |
 | 3 | [Reconstruction](https://github.com/vinits5/learning3d#use-of-point-completion-network) | Point Completion Network (PCN) |
 | 4 | [Registration](https://github.com/vinits5/learning3d#use-of-registration-networks) | PointNetLK, PCRNet, DCP, PRNet, RPM-Net, DeepGMR |
 | 5 | [Flow Estimation](https://github.com/vinits5/learning3d#use-of-flow-estimation-network) | FlowNet3D |
-| 6 | [Inlier Estimation](https://github.com/vinits5/learning3d#use-of-inlier-estimation-network-masknet) | MaskNet, MaskNet++ | 
+| 6 | [Inlier Estimation](https://github.com/vinits5/learning3d#use-of-inlier-estimation-network-masknet) | MaskNet, [MaskNet++](https://github.com/zhouruqin/MaskNet2) | 
 
 ## Available Pretrained Models
 1. PointNet
@@ -41,6 +57,7 @@ Learning3D is an open-source library that supports the development of deep learn
 10. PointConv (Download from this [link](https://github.com/DylanWusee/pointconv_pytorch/blob/master/checkpoints/checkpoint.pth))
 11. MaskNet
 12. MaskNet++ / MaskNet2
+13. CurveNet
 
 ## Available Datasets
 1. ModelNet40
@@ -56,7 +73,8 @@ Learning3D is an open-source library that supports the development of deep learn
 1. Ubuntu 16.04
 2. Ubuntu 18.04
 3. Ubuntu 20.04.6
-3. Linux Mint
+4. Linux Mint
+5. macOS Sequoia 15.3.1
 
 ### Requirements
 1. CUDA 10.0 or higher
@@ -242,4 +260,5 @@ PointConv variable is a class. Users can use it to create a sub-class to overrid
 13. [DeepGMR:](https://arxiv.org/abs/2008.09088) Learning Latent Gaussian Mixture Models for Registration
 14. [CMU:](https://arxiv.org/pdf/2010.16085.pdf) Correspondence Matrices are Underrated
 15. [MaskNet:](https://arxiv.org/pdf/2010.09185.pdf) A Fully-Convolutional Network to Estimate Inlier Points
-16. [MaskNet++:](https://www.sciencedirect.com/science/article/abs/pii/S0097849322000085) Inlier/outlier identification for two point clouds
+16. [MaskNet++:](https://www.sciencedirect.com/science/article/abs/pii/S0097849322000085) Inlier/outlier identification for two point clouds
+17. [CurveNet:](https://github.com/tiangexiang/CurveNet) Walk in the Cloud: Learning Curves for Point Clouds Shape Analysis

{learning3d-0.1.0/src → learning3d-0.2.1}/learning3d/data_utils/dataloaders.py

@@ -16,8 +16,9 @@ from scipy.spatial.distance import minkowski
 from scipy.spatial import cKDTree
 from torch.utils.data import Dataset
 
-def download_modelnet40(root_dir):
-	DATA_DIR = os.path.join(root_dir, 'data')
+def download_modelnet40():
+	BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+	DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
 	if not os.path.exists(DATA_DIR):
 		os.mkdir(DATA_DIR)
 	if not os.path.exists(os.path.join(DATA_DIR, 'modelnet40_ply_hdf5_2048')):
@@ -27,10 +28,11 @@ def download_modelnet40(root_dir):
 		os.system('mv %s %s' % (zipfile[:-4], DATA_DIR))
 		os.system('rm %s' % (zipfile))
 
-def load_data(root_dir, train, use_normals):
+def load_data(train, use_normals):
 	if train: partition = 'train'
 	else: partition = 'test'
-	DATA_DIR = os.path.join(root_dir, 'data')
+	BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+	DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
 	all_data = []
 	all_label = []
 	for h5_name in glob.glob(os.path.join(DATA_DIR, 'modelnet40_ply_hdf5_2048', 'ply_data_%s*.h5' % partition)):
@@ -184,15 +186,13 @@ class ModelNet40Data(Dataset):
 		self,
 		train=True,
 		num_points=1024,
-		download=False,
-		root_dir='./',
+		download=True,
 		randomize_data=False,
 		use_normals=False
 	):
 		super(ModelNet40Data, self).__init__()
-		self.root_dir = root_dir
-		if download: download_modelnet40(root_dir=root_dir)
-		self.data, self.labels = load_data(root_dir, train, use_normals)
+		if download: download_modelnet40()
+		self.data, self.labels = load_data(train, use_normals)
 		if not train: self.shapes = self.read_classes_ModelNet40()
 		self.num_points = num_points
 		self.randomize_data = randomize_data
@@ -218,7 +218,8 @@ class ModelNet40Data(Dataset):
 		return self.shapes[label]
 
 	def read_classes_ModelNet40(self):
-		DATA_DIR = os.path.join(self.root_dir, 'data')
+		BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+		DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
 		file = open(os.path.join(DATA_DIR, 'modelnet40_ply_hdf5_2048', 'shape_names.txt'), 'r')
 		shape_names = file.read()
 		shape_names = np.array(shape_names.split('\n')[:-1])
@@ -226,7 +227,7 @@ class ModelNet40Data(Dataset):
 
 
 class ClassificationData(Dataset):
-	def __init__(self, data_class):
+	def __init__(self, data_class=ModelNet40Data()):
 		super(ClassificationData, self).__init__()
 		self.set_class(data_class)
 
@@ -247,7 +248,7 @@ class ClassificationData(Dataset):
 
 
 class RegistrationData(Dataset):
-	def __init__(self, algorithm, data_class, partial_source=False, partial_template=False, noise=False, additional_params={}):
+	def __init__(self, algorithm, 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
@@ -361,9 +362,10 @@ class FlowData(Dataset):
 
 
 class SceneflowDataset(Dataset):
-	def __init__(self, root_dir, npoints=1024, root='', partition='train'):
+	def __init__(self, npoints=1024, root='', partition='train'):
 		if root == '':
-			DATA_DIR = os.path.join(root_dir, 'data')
+			BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+			DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
 			root = os.path.join(DATA_DIR, 'data_processed_maxcut_35_20k_2k_8192')
 			if not os.path.exists(root): 
 				print("To download dataset, click here: https://drive.google.com/file/d/1CMaxdt-Tg1Wct8v8eGNwuT7qRSIyJPY-/view")

learning3d-0.2.1/learning3d/examples/test_curvenet.py

@@ -0,0 +1,118 @@
+import open3d as o3d
+import argparse
+import os
+import sys
+import logging
+import numpy
+import numpy as np
+import torch
+import torch.utils.data
+import torchvision
+from torch.utils.data import DataLoader
+from tensorboardX import SummaryWriter
+from tqdm import tqdm
+
+# Only if the files are in example folder.
+BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+if BASE_DIR[-8:] == 'examples':
+	sys.path.append(os.path.join(BASE_DIR, os.pardir))
+	os.chdir(os.path.join(BASE_DIR, os.pardir))
+	
+from learning3d.models import CurveNet
+from learning3d.data_utils import ClassificationData, ModelNet40Data
+
+def display_open3d(template):
+	template_ = o3d.geometry.PointCloud()
+	template_.points = o3d.utility.Vector3dVector(template)
+	# template_.paint_uniform_color([1, 0, 0])
+	o3d.visualization.draw_geometries([template_])
+
+def test_one_epoch(device, model, test_loader, testset):
+	model.eval()
+	test_loss = 0.0
+	pred  = 0.0
+	count = 0
+	for i, data in enumerate(tqdm(test_loader)):
+		points, target = data
+		target = target[:,0]
+
+		points = points.to(device)
+		target = target.to(device)
+
+		output = model(points)
+		loss_val = torch.nn.functional.nll_loss(
+			torch.nn.functional.log_softmax(output, dim=1), target, size_average=False)
+		print("Ground Truth Label: ", testset.get_shape(target[0].item()))
+		print("Predicted Label:    ", testset.get_shape(torch.argmax(output[0]).item()))
+		display_open3d(points.detach().cpu().numpy()[0])
+
+		test_loss += loss_val.item()
+		count += output.size(0)
+
+		_, pred1 = output.max(dim=1)
+		ag = (pred1 == target)
+		am = ag.sum()
+		pred += am.item()
+
+	test_loss = float(test_loss)/count
+	accuracy = float(pred)/count
+	return test_loss, accuracy
+
+def test(args, model, test_loader, testset):
+	test_loss, test_accuracy = test_one_epoch(args.device, model, test_loader, testset)
+	print("Accuracy: ", test_accuracy*100)
+
+def options():
+	parser = argparse.ArgumentParser(description='Point Cloud Registration')
+	parser.add_argument('--dataset_path', type=str, default='ModelNet40',
+						metavar='PATH', help='path to the input dataset') # like '/path/to/ModelNet40'
+	parser.add_argument('--eval', type=bool, default=False, help='Train or Evaluate the network.')
+
+	# settings for input data
+	parser.add_argument('--dataset_type', default='modelnet', choices=['modelnet', 'shapenet2'],
+						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)')
+
+	# settings for CurveNet
+	parser.add_argument('-j', '--workers', default=4, type=int,
+						metavar='N', help='number of data loading workers (default: 4)')
+	parser.add_argument('-b', '--batch_size', default=32, type=int,
+						metavar='N', help='mini-batch size (default: 32)')
+	parser.add_argument('--num_classes', default=40, type=int,
+						metavar='K', help='number of classes to be predicted')
+
+	# settings for on training
+	parser.add_argument('--pretrained', default='learning3d/pretrained/exp_curvenet/models/model.t7', type=str,
+						metavar='PATH', help='path to pretrained model file (default: null (no-use))')
+	parser.add_argument('--device', default='cuda:0', type=str,
+						metavar='DEVICE', help='use CUDA if available')
+
+	args = parser.parse_args()
+	return args
+
+def main():
+	args = options()
+	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
+	
+	testset = ClassificationData(ModelNet40Data(train=False))
+	test_loader = DataLoader(testset, batch_size=args.batch_size, shuffle=False, drop_last=False, num_workers=args.workers)
+
+	if not torch.cuda.is_available():
+		args.device = 'cpu'
+	args.device = torch.device(args.device)
+
+	# Create PointNet Model.
+	model = CurveNet(num_classes=args.num_classes, k=20)
+
+	if args.pretrained:
+		assert os.path.isfile(args.pretrained)
+		weights = torch.load(args.pretrained, map_location='cpu')
+		weights = {k[7:]: v for k, v in weights.items()}
+		model.load_state_dict(weights)
+	model.to(args.device)
+
+	test(args, model, test_loader, testset)
+
+if __name__ == '__main__':
+	main()

{learning3d-0.1.0/src → learning3d-0.2.1}/learning3d/examples/test_dcp.py

@@ -88,8 +88,6 @@ 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'],
@@ -116,8 +114,8 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 	
-	trainset = RegistrationData('DCP', ModelNet40Data(train=True, root_dir=args.root_dir))
-	testset = RegistrationData('DCP', ModelNet40Data(train=False, root_dir=args.root_dir))
+	trainset = RegistrationData('DCP', ModelNet40Data(train=True))
+	testset = RegistrationData('DCP', ModelNet40Data(train=False))
 	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)
 
@@ -132,7 +130,7 @@ def main():
 
 	if args.pretrained:
 		assert os.path.isfile(args.pretrained)
-		model.load_state_dict(torch.load(args.pretrained), strict=False)
+		model.load_state_dict(torch.load(args.pretrained, map_location='cpu'), strict=False)
 	model.to(args.device)
 
 	test(args, model, test_loader)

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

@@ -103,8 +103,6 @@ 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,
@@ -123,8 +121,8 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 	
-	trainset = RegistrationData('DeepGMR', ModelNet40Data(train=True, root_dir=args.root_dir))
-	testset = RegistrationData('DeepGMR', ModelNet40Data(train=False, root_dir=args.root_dir))
+	trainset = RegistrationData('DeepGMR', ModelNet40Data(train=True))
+	testset = RegistrationData('DeepGMR', ModelNet40Data(train=False))
 	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)
 
@@ -137,7 +135,7 @@ def main():
 
 	if args.pretrained:
 		assert os.path.isfile(args.pretrained)
-		model.load_state_dict(torch.load(args.pretrained), strict=False)
+		model.load_state_dict(torch.load(args.pretrained, map_location='cpu'), strict=False)
 	model.to(args.device)
 
 	test(args, model, test_loader)

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

@@ -117,8 +117,6 @@ 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,
@@ -139,7 +137,7 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points, root_dir=args.root_dir),
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points),
 									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.1.0/src → learning3d-0.2.1}/learning3d/examples/test_masknet2.py

@@ -120,8 +120,6 @@ 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,
@@ -142,7 +140,7 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = True
 
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points, root_dir=args.root_dir),
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, num_points=args.num_points),
 									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.1.0/src → learning3d-0.2.1}/learning3d/examples/test_pcn.py

@@ -70,8 +70,6 @@ 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,
@@ -97,8 +95,8 @@ def main():
 	args = options()
 	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
 
-	trainset = ClassificationData(ModelNet40Data(train=True, root_dir=args.root_dir))
-	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
+	trainset = ClassificationData(ModelNet40Data(train=True))
+	testset = ClassificationData(ModelNet40Data(train=False))
 	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.1.0/src → learning3d-0.2.1}/learning3d/examples/test_pcrnet.py

@@ -74,8 +74,6 @@ 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,
@@ -99,7 +97,7 @@ def options():
 def main():
 	args = options()
 
-	testset = RegistrationData('PCRNet', ModelNet40Data(train=False, root_dir=args.root_dir))
+	testset = RegistrationData('PCRNet', ModelNet40Data(train=False))
 	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.1.0/src → learning3d-0.2.1}/learning3d/examples/test_pnlk.py

@@ -74,8 +74,6 @@ 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,
@@ -100,7 +98,7 @@ def options():
 def main():
 	args = options()
 
-	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False, root_dir=args.root_dir))
+	testset = RegistrationData('PointNetLK', ModelNet40Data(train=False))
 	test_loader = DataLoader(testset, batch_size=8, shuffle=False, drop_last=False, num_workers=args.workers)
 
 	if not torch.cuda.is_available():

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

@@ -73,8 +73,6 @@ 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'],
@@ -101,7 +99,7 @@ def main():
 	args = options()
 	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
 	
-	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
+	testset = ClassificationData(ModelNet40Data(train=False))
 	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.1.0/src → learning3d-0.2.1}/learning3d/examples/test_pointnet.py

@@ -73,8 +73,6 @@ 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'],
@@ -101,7 +99,7 @@ def main():
 	args = options()
 	args.dataset_path = os.path.join(os.getcwd(), os.pardir, os.pardir, 'ModelNet40', 'ModelNet40')
 	
-	testset = ClassificationData(ModelNet40Data(train=False, root_dir=args.root_dir))
+	testset = ClassificationData(ModelNet40Data(train=False))
 	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.1.0/src → learning3d-0.2.1}/learning3d/examples/test_prnet.py

@@ -79,8 +79,6 @@ 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,
@@ -104,8 +102,8 @@ def main():
 	args = options()
 	torch.backends.cudnn.deterministic = 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)
+	trainset = RegistrationData('PRNet', ModelNet40Data(train=True), partial_source=True, partial_template=True)
+	testset = RegistrationData('PRNet', ModelNet40Data(train=False), 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)
 
@@ -119,7 +117,7 @@ def main():
 
 	if args.pretrained:
 		assert os.path.isfile(args.pretrained)
-		model.load_state_dict(torch.load(args.pretrained), strict=False)
+		model.load_state_dict(torch.load(args.pretrained, map_location='cpu'), strict=False)
 	model.to(args.device)
 
 	test(args, model, test_loader)