Update ModelNetDataLoader.py

nidhi23ug · Apr 5, 2021 · 8e26b85 · 8e26b85
1 parent f7a8a93
commit 8e26b85
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/data_utils/ModelNetDataLoader.py b/data_utils/ModelNetDataLoader.py
@@ -115,7 +115,6 @@ def __len__(self):
     def _get_item(self, index):
         if self.process_data:
             point_set, label = self.list_of_points[index], self.list_of_labels[index]
-            point_set[:, 0:3] = pc_normalize(point_set[:, 0:3])
         else:
             fn = self.datapath[index]
             cls = self.classes[self.datapath[index][0]]
@@ -126,7 +125,8 @@ def _get_item(self, index):
                 point_set = farthest_point_sample(point_set, self.npoints)
             else:
                 point_set = point_set[0:self.npoints, :]
-
+
+        point_set[:, 0:3] = pc_normalize(point_set[:, 0:3])
         if not self.use_normals:
             point_set = point_set[:, 0:3]