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)