wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-08-12 11:40:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

memcpy is super slow in cuda kernels....

Browse Source
This commit is contained in:
erikwijmans
2018-01-29 18:22:27 -05:00
parent 009c30e5e3
commit c4ddd6b099
7 changed files with 30 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/Indoor3DSemSegLoader.py
+3 -2
View File
@@ -5,7 +5,8 @@ import os, sys, h5py, subprocess, shlex
def _get_data_files(list_filename):
return [line.rstrip() for line in open(list_filename)]
with open(list_filename) as f:
return [line.rstrip() for line in f]
def _load_data_file(name):
@@ -76,7 +77,7 @@ 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(
current_points = torch.from_numpy(self.points[idx, pt_idxs]).type(
torch.FloatTensor
)
current_labels = torch.from_numpy(self.labels[idx, pt_idxs]).type(