wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-06-28 16:10:08 +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/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: