File based header detection, collate_per_PC training.

vis/show_seg_res.py

@@ -5,7 +5,7 @@ import sys
 import os
 sys.path.append(os.path.dirname(os.path.abspath(__file__)) + '/../')  # add project root directory
 
-from dataset.shapenet import ShapeNetPartSegDataset
+from dataset.shapenet import PredictNetPartSegDataset, ShapeNetPartSegDataset
 from model.pointnet2_part_seg import PointNet2PartSegmentNet
 import torch_geometric.transforms as GT
 import torch
@@ -28,7 +28,7 @@ if __name__ == '__main__':
     print('Construct dataset ..')
     test_transform = GT.Compose([GT.NormalizeScale(),])
 
-    test_dataset = ShapeNetPartSegDataset(
+    test_dataset = PredictNetPartSegDataset(
         root_dir=opt.dataset,
         collate_per_segment=False,
         train=False,
@@ -128,12 +128,12 @@ if __name__ == '__main__':
             print('View gt labels ..')
             view_points_labels(points, gt_labels)
 
-        if True:
+        if False:
             print('View diff labels ..')
             print(diff_labels)
             view_points_labels(points, diff_labels)
 
-        if True:
+        if False:
             print('View pred labels ..')
             print(pred_labels)
             view_points_labels(points, pred_labels)