learning3d 0.0.3__tar.gz → 0.0.4__tar.gz

{learning3d-0.0.3/src/learning3d.egg-info → learning3d-0.0.4}/PKG-INFO

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

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

{learning3d-0.0.3 → learning3d-0.0.4}/src/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=None,
 		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 → learning3d-0.0.4/src/learning3d.egg-info}/PKG-INFO

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