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

clean up

Browse Source
This commit is contained in:
wassname
2018-04-10 13:40:05 +08:00
parent 0a6b08b83f
commit 9046814dc6
2 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/pointnet2_msg_sem.py
-3
View File
@@ -62,7 +62,6 @@ class Pointnet2MSG(nn.Module):
radii=[0.1, 0.2],
nsamples=[16, 32],
mlps=[[c_in, 64, 64, 128], [c_in, 64, 96, 128]],
# use_xyz=use_xyz
)
)
c_out_1 = 128 + 128
@@ -74,7 +73,6 @@ class Pointnet2MSG(nn.Module):
radii=[0.2, 0.4],
nsamples=[16, 32],
mlps=[[c_in, 128, 196, 256], [c_in, 128, 196, 256]],
# use_xyz=use_xyz
)
)
c_out_2 = 256 + 256
@@ -86,7 +84,6 @@ class Pointnet2MSG(nn.Module):
radii=[0.4, 0.8],
nsamples=[16, 32],
mlps=[[c_in, 256, 256, 512], [c_in, 256, 384, 512]],
# use_xyz=use_xyz
)
)
c_out_3 = 512 + 512
utils/pointnet2_modules.py
+1 -1
View File
@@ -22,7 +22,7 @@ class _PointnetSAModuleBase(nn.Module):
----------
xyz : torch.Tensor
(B, N, 3) tensor of the xyz coordinates of the points
point : torch.Tensor
points : torch.Tensor
(B, N, C) tensor of the descriptors of the the points
Returns