x

fast_pointnet_v2.py

@@ -158,7 +158,7 @@ class PatchDataset(Dataset):
     Updated for 11D patches.
     """
     
-    def __init__(self, dataset_dir: str, max_points: int = 1024, augment: bool = False):
+    def __init__(self, dataset_dir: str, max_points: int = 1024, augment: bool = True):
         self.dataset_dir = dataset_dir
         self.max_points = max_points
         self.augment = augment