adjust threshold for viz

mapper/utils/viz_2d.py

@@ -83,8 +83,9 @@ def one_hot_argmax_to_rgb(y, num_class):
     class_colors = class_colors.values()
     class_colors = [torch.tensor(x).float() for x in class_colors]
 
-    argmaxed = torch.argmax((y > 0.5).float(), dim=1) # Take argmax
-    argmaxed[torch.all(y <= 0.5, dim=1)] = num_class
+    threshold = 0.25
+    argmaxed = torch.argmax((y > threshold).float(), dim=1) # Take argmax
+    argmaxed[torch.all(y <= threshold, dim=1)] = num_class
     # print(argmaxed.shape)
 
     seg_rgb = torch.ones(