wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-06-27 16:00:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Possible bug fixes

Browse Source
This commit is contained in:
erikwijmans
2018-03-06 20:55:40 -05:00
parent f0b92f7018
commit 5fbc7effa6
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/Indoor3DSemSegLoader.py
+4 -6
View File
@@ -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
data/ModelNet40Loader.py
+1 -1
View File
@@ -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: