Update the color for the segmentation.