Cmap for segmentation

For the unet_learner, I was trying to pass a different color map for the label into segmentation results using


However, the output color map of the label remain the same. It seems that in the, cmap is passed to plotting the image x, instead of passed in **kwargs to ImageSegment y.

How may I change the label color map?


Alex: Do you have an answer for this issue? I am trying to do the same.

OR do you which value turn the mask color to “black”? I try zero (0) but that only darken it, but not darken enough.