File based header detection, collate_per_PC training.

vis/show_seg_res.py

@@ -122,9 +122,10 @@ if __name__ == '__main__':
         diff_labels = label_diff(pred_labels, gt_labels)
 
         print('mIoU: ', compute_mIoU(pred_labels, gt_labels))
+        print(pred_labels)
 
         # View result
-        if True:
+        if False:
             print('View gt labels ..')
             view_points_labels(points, gt_labels)
 

vis/view.py

@@ -37,7 +37,6 @@ def view_points(points, colors=None):
     o3d.draw_geometries([cloud])
 
 
-
 def label2color(labels):
     '''
     labels: np.ndarray with shape (n, )