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

fix tests

Browse Source
This commit is contained in:
wassname
2018-04-10 11:46:28 +08:00
parent 96f46514d6
commit 20af9060dd
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
data/ModelNet40Loader.py
+1 -1
View File
@@ -98,7 +98,7 @@ if __name__ == "__main__":
transforms = transforms.Compose([ transforms = transforms.Compose([
d_utils.PointcloudToTensor(), d_utils.PointcloudToTensor(),
d_utils.PointcloudRotate(x_axis=True), d_utils.PointcloudRotate(axis=np.array([1,0,0])),
d_utils.PointcloudScale(), d_utils.PointcloudScale(),
d_utils.PointcloudTranslate(), d_utils.PointcloudTranslate(),
d_utils.PointcloudJitter() d_utils.PointcloudJitter()
utils/pointnet2_modules.py
+2 -2
View File
@@ -214,8 +214,8 @@ if __name__ == "__main__":
from torch.autograd import Variable from torch.autograd import Variable
torch.manual_seed(1) torch.manual_seed(1)
torch.cuda.manual_seed_all(1) torch.cuda.manual_seed_all(1)
xyz = Variable(torch.randn(2, 10, 3).cuda(), requires_grad=True) xyz = Variable(torch.randn(2, 9, 3).cuda(), requires_grad=True)
xyz_feats = Variable(torch.randn(2, 10, 6).cuda(), requires_grad=True) xyz_feats = Variable(torch.randn(2, 9, 6).cuda(), requires_grad=True)
test_module = PointnetSAModuleMSG( test_module = PointnetSAModuleMSG(
npoint=2, radii=[5.0, 10.0], nsamples=[6, 3], mlps=[[9, 3], [9, 6]] npoint=2, radii=[5.0, 10.0], nsamples=[6, 3], mlps=[[9, 3], [9, 6]]