Skip to content
Snippets Groups Projects
Commit 308cc51c authored by Alexandre Chapin's avatar Alexandre Chapin :race_car:
Browse files

Fix issue on computing render

parent 7d67128a
No related branches found
No related tags found
No related merge requests found
......@@ -285,7 +285,7 @@ def visualize(model, vis_data, out_dir, render_args, fabric, config):
columns.append((f'pred seg {angle_deg}°', pred_seg.argmax(-1).numpy(), 'clustering'))
if i == 0:
ari = compute_adjusted_rand_index(
input_mask.flatten(1, 2).transpose(1, 2)[:, 1:],
input_mask.cpu().flatten(1, 2).transpose(1, 2)[:, 1:],
pred_seg.flatten(1, 2).transpose(1, 2))
row_labels = ['2D Fg-ARI={:.1f}'.format(x.item() * 100) for x in ari]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment