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

Updates to data augmentation

Browse Source
This commit is contained in:
erikwijmans
2018-02-22 20:08:56 -05:00
parent 2730bbf963
commit 0515ba5c69
3 changed files with 154 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train_cls.py
+6 -3
View File
@@ -13,7 +13,7 @@ from models import Pointnet2ClsMSG as Pointnet
from models.Pointnet2Cls import model_fn_decorator
from data import ModelNet40Cls
import utils.pytorch_utils as pt_utils
import utils.data_utils as d_utils
import data.data_utils as d_utils
import argparse
torch.backends.cudnn.enabled = True
@@ -85,9 +85,12 @@ if __name__ == "__main__":
transforms = transforms.Compose([
d_utils.PointcloudToTensor(),
d_utils.PointcloudRotate(x_axis=True, z_axis=True),
d_utils.PointcloudScale(),
d_utils.PointcloudRotate(),
d_utils.PointcloudRotatePerturbation(),
d_utils.PointcloudTranslate(),
d_utils.PointcloudJitter()
d_utils.PointcloudJitter(),
d_utils.PointcloudRandomInputDropout()
])
test_set = ModelNet40Cls(