learning3d 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

learning3d/data_utils/dataloaders.py

@@ -16,9 +16,8 @@ from scipy.spatial.distance import minkowski
 from scipy.spatial import cKDTree
 from torch.utils.data import Dataset
 
-def download_modelnet40():
-	BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-	DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
+def download_modelnet40(root_dir):
+	DATA_DIR = os.path.join(root_dir, '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')):
@@ -28,11 +27,10 @@ def download_modelnet40():
 		os.system('mv %s %s' % (zipfile[:-4], DATA_DIR))
 		os.system('rm %s' % (zipfile))
 
-def load_data(train, use_normals):
+def load_data(root_dir, train, use_normals):
 	if train: partition = 'train'
 	else: partition = 'test'
-	BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-	DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
+	DATA_DIR = os.path.join(root_dir, '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)):
@@ -186,13 +184,14 @@ class ModelNet40Data(Dataset):
 		self,
 		train=True,
 		num_points=1024,
-		download=True,
+		root_dir='./',
 		randomize_data=False,
 		use_normals=False
 	):
 		super(ModelNet40Data, self).__init__()
-		if download: download_modelnet40()
-		self.data, self.labels = load_data(train, use_normals)
+		self.root_dir = root_dir
+		if root_dir is not None: download_modelnet40(root_dir=root_dir)
+		self.data, self.labels = load_data(root_dir, train, use_normals)
 		if not train: self.shapes = self.read_classes_ModelNet40()
 		self.num_points = num_points
 		self.randomize_data = randomize_data
@@ -218,8 +217,7 @@ class ModelNet40Data(Dataset):
 		return self.shapes[label]
 
 	def read_classes_ModelNet40(self):
-		BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-		DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
+		DATA_DIR = os.path.join(self.root_dir, '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])
@@ -362,10 +360,9 @@ class FlowData(Dataset):
 
 
 class SceneflowDataset(Dataset):
-	def __init__(self, npoints=1024, root='', partition='train'):
+	def __init__(self, root_dir, npoints=1024, root='', partition='train'):
 		if root == '':
-			BASE_DIR = os.path.dirname(os.path.abspath(__file__))
-			DATA_DIR = os.path.join(BASE_DIR, os.pardir, 'data')
+			DATA_DIR = os.path.join(root_dir, '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.0.3.dist-info → learning3d-0.0.5.dist-info}/METADATA

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

@@ -1,6 +1,6 @@
 learning3d/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 learning3d/data_utils/__init__.py,sha256=iYAVh0FThnVlG42QIgmDYrC3NGVYuzKX8s1oRqAI1YU,261
-learning3d/data_utils/dataloaders.py,sha256=kb0wsLlMN7sB-CS_4BGSprSaZBwkUNYxS5iwUdD6JJM,14871
+learning3d/data_utils/dataloaders.py,sha256=xoH9gzfcCUOqcRL3G_Qjglt-vuNooIW2oqgRDhT3nv8,14703
 learning3d/data_utils/user_data.py,sha256=ADDGeCUCr6TcXhcxvAFncIeLO71xoRHYi4H418ktvQs,4828
 learning3d/examples/test_dcp.py,sha256=o1hgl22b2xhqqYe4_k7NlNo3-rIMmSzzssCI74DCfoE,5606
 learning3d/examples/test_deepgmr.py,sha256=pb-gRE6YLLaPFmiBXQjaE-B-J314QY4RkNwLHx65bcU,5597
@@ -73,8 +73,8 @@ learning3d/utils/pointconv_util.py,sha256=kJxGztai7X15YsGuorMOc50SPtj_k1yfkP4XCT
 learning3d/utils/ppfnet_util.py,sha256=HEoxkgUBlawKZLWspfQm3caWUyAMIrW-ECtStNYbe2Y,7989
 learning3d/utils/svd.py,sha256=yCYQt2SKqeIzCBnBEr_8xFR79m4fIoNVFnp77epn1dM,1936
 learning3d/utils/transformer.py,sha256=UDgJvnh7ekWyijaAn-a3ckeFeMxlK_chXzWlhAGDiPM,8974
-learning3d-0.0.3.dist-info/LICENSE,sha256=3qY3_NeQIvalbLlsHFtOfuUKjs_U2k6u7rf6YVx6ac0,1098
-learning3d-0.0.3.dist-info/METADATA,sha256=RjRxCmF_3xda4kPWDCvP2QjyPSysHZdNm3PY42-Jlhg,15813
-learning3d-0.0.3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-learning3d-0.0.3.dist-info/top_level.txt,sha256=nTmYW8NhbNV1_15DGNpl_OvvSFtQP98sy3qrrHr0eLo,11
-learning3d-0.0.3.dist-info/RECORD,,
+learning3d-0.0.5.dist-info/LICENSE,sha256=3qY3_NeQIvalbLlsHFtOfuUKjs_U2k6u7rf6YVx6ac0,1098
+learning3d-0.0.5.dist-info/METADATA,sha256=gdmOxHdGi3GXBcPxKYWAdHvr9S8tSVNjMN2W_B_aiww,15813
+learning3d-0.0.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+learning3d-0.0.5.dist-info/top_level.txt,sha256=nTmYW8NhbNV1_15DGNpl_OvvSFtQP98sy3qrrHr0eLo,11
+learning3d-0.0.5.dist-info/RECORD,,