Now with included sorting per cloud variation - Some korrektions

dataset/shapenet.py

@@ -137,7 +137,7 @@ class CustomShapeNet(InMemoryDataset):
                     if self.collate_per_element:
                         datasets[data_folder].append(data)
                     if self.has_variations:
-                        cloud_variations[int(element.name.split('_')[0])].append(data)
+                        cloud_variations[int(os.path.split(element)[-1].split('_')[0])].append(data)
             if not self.collate_per_element:
                 if self.has_variations:
                     for variation in cloud_variations.keys():