File based header detection, collate_per_PC training.

This commit is contained in:
Si11ium 2019-08-01 18:17:46 +02:00
parent b7d127e840
commit ff117ea2f2
2 changed files with 2 additions and 2 deletions

View File

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

View File

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