Update ShapeNetDataLoader.py

fixed the bug (yanx27#105)
nidhi23ug · Apr 28, 2021 · ba4c5cf · ba4c5cf
1 parent bba1f61
commit ba4c5cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_utils/ShapeNetDataLoader.py b/data_utils/ShapeNetDataLoader.py
@@ -88,7 +88,7 @@ def __init__(self,root = './data/shapenetcore_partanno_segmentation_benchmark_v0
 
     def __getitem__(self, index):
         if index in self.cache:
-            ppoint_set, cls, seg = self.cache[index]
+            point_set, cls, seg = self.cache[index]
         else:
             fn = self.datapath[index]
             cat = self.datapath[index][0]