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

Updates and some refactoring

Browse Source
This commit is contained in:
erikwijmans
2017-12-26 19:49:52 -05:00
parent dc4e2b0db3
commit 803d7e1fc6
10 changed files with 211 additions and 226 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/pointnet2_modules.py
+1 -1
View File
@@ -142,7 +142,7 @@ class PointnetSAModule(nn.Module):
new_xyz = pointnet2_utils.gather_points(
xyz, pointnet2_utils.furthest_point_sample(xyz, self.npoint))
else:
new_xyz = xyz.new([[[0, 0, 0]]]).expand(xyz.size(0), 1, 3)
new_xyz = xyz.data.new([[[0, 0, 0]]]).expand(xyz.size(0), 1, 3)
new_points = self.grouper(xyz, new_xyz,
points) # (B, npoint, nsample, 3 + C)