updated codes for classification

nidhi23ug · Mar 20, 2021 · bfd5b39 · bfd5b39
1 parent e101947
commit bfd5b39
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 5 deletions.
diff --git a/data_utils/ModelNetDataLoader.py b/data_utils/ModelNetDataLoader.py
@@ -130,7 +130,6 @@ def _get_item(self, index):
         if not self.use_normals:
             point_set = point_set[:, 0:3]
 
-
         return point_set, label[0]
 
     def __getitem__(self, index):

diff --git a/test_classification.py b/test_classification.py
@@ -88,8 +88,7 @@ def log_string(str):
 
     '''DATA LOADING'''
     log_string('Load dataset ...')
-    # data_path = 'data/modelnet40_normal_resampled/'
-    data_path = '/data/dataset/modelnet40_normal_resampled/'
+    data_path = 'data/modelnet40_normal_resampled/'
 
     test_dataset = ModelNetDataLoader(root=data_path, args=args, split='test', process_data=False)
     testDataLoader = torch.utils.data.DataLoader(test_dataset, batch_size=args.batch_size, shuffle=False, num_workers=10)

diff --git a/train_classification.py b/train_classification.py
@@ -110,8 +110,7 @@ def log_string(str):
 
     '''DATA LOADING'''
     log_string('Load dataset ...')
-    # data_path = 'data/modelnet40_normal_resampled/'
-    data_path = '/data/dataset/modelnet40_normal_resampled/'
+    data_path = 'data/modelnet40_normal_resampled/'
 
     train_dataset = ModelNetDataLoader(root=data_path, args=args, split='train', process_data=args.process_data)
     test_dataset = ModelNetDataLoader(root=data_path, args=args, split='test', process_data=args.process_data)