Update mono/utils/transform.py

mono/utils/transform.py

@@ -399,7 +399,12 @@ def gray_to_colormap(img, cmap='rainbow'):
 
     img[img<0] = 0
     mask_invalid = img < 1e-10
-    img = img / (img.max() + 1e-8)
+    #img = img / (img.max() + 1e-8)
+
+    img_ = img.flatten()
+    max_value = np.percentile(img_, q=98)
+    img = img / (max_value + 1e-8)
+    
     norm = matplotlib.colors.Normalize(vmin=0, vmax=1.1)
     cmap_m = matplotlib.cm.get_cmap(cmap)
     map = matplotlib.cm.ScalarMappable(norm=norm, cmap=cmap_m)