merged

vis/show_seg_res.py

@@ -28,10 +28,10 @@ if __name__ == '__main__':
     print('Construct dataset ..')
     test_transform = GT.Compose([GT.NormalizeScale(),])
 
-    test_dataset = PredictNetPartSegDataset(
+    test_dataset = ShapeNetPartSegDataset(
         root_dir=opt.dataset,
         collate_per_segment=False,
-        train=False,
+        mode='predict',
         transform=test_transform,
         npoints=opt.npoints
     )