Pointnet_Example/utils.py

@@ -60,7 +60,8 @@ class PointCloudDataSet():
             points = load_las_file_to_numpy(self.all_files[idx])
             if self.transform:
                 points = self.transform(points)
-            label = self.all_labels.iloc[idx, 1:].species
+            file_id = int(self.all_files[idx].split('/')[-1].split('_')[-1].split('.')[0])
+            label = self.all_labels.iloc[file_id, 1:].species
         except:
             print(
                 f"Problem with reading file number {idx} at {self.all_files[idx]}")