diff --git a/vis/show_seg_res.py b/vis/show_seg_res.py index 6de9a6a..7c75122 100644 --- a/vis/show_seg_res.py +++ b/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) diff --git a/vis/view.py b/vis/view.py index b99f55f..8da396a 100644 --- a/vis/view.py +++ b/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, )