diff --git a/data/Indoor3DSemSegLoader.py b/data/Indoor3DSemSegLoader.py index e9a5b47..3c12759 100644 --- a/data/Indoor3DSemSegLoader.py +++ b/data/Indoor3DSemSegLoader.py @@ -77,12 +77,10 @@ class Indoor3DSemSeg(data.Dataset): pt_idxs = np.arange(0, self.num_points) np.random.shuffle(pt_idxs) - current_points = torch.from_numpy(self.points[idx, pt_idxs]).type( - torch.FloatTensor - ) - current_labels = torch.from_numpy(self.labels[idx, pt_idxs]).type( - torch.LongTensor - ) + current_points = torch.from_numpy(self.points[idx, pt_idxs].copy() + ).type(torch.FloatTensor) + current_labels = torch.from_numpy(self.labels[idx, pt_idxs].copy() + ).type(torch.LongTensor) return current_points, current_labels diff --git a/data/ModelNet40Loader.py b/data/ModelNet40Loader.py index ab8a767..150d09b 100644 --- a/data/ModelNet40Loader.py +++ b/data/ModelNet40Loader.py @@ -68,7 +68,7 @@ class ModelNet40Cls(data.Dataset): pt_idxs = np.arange(0, self.actual_number_of_points) np.random.shuffle(pt_idxs) - current_points = self.points[idx, pt_idxs, :] + current_points = self.points[idx, pt_idxs].copy() label = torch.from_numpy(self.labels[idx]).type(torch.LongTensor) if self.transforms is not None: