From 5fbc7effa613322e6d0564d4c2d28a85ce190c65 Mon Sep 17 00:00:00 2001 From: erikwijmans Date: Tue, 6 Mar 2018 20:55:40 -0500 Subject: [PATCH] Possible bug fixes --- data/Indoor3DSemSegLoader.py | 10 ++++------ data/ModelNet40Loader.py | 2 +- 2 files changed, 5 insertions(+), 7 deletions(-) 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: