code: def compute_colors_for_labels(self, labels):
          
          colors = labels[:, None] * self.palette
          colors = (colors % 255).numpy().astype("uint8")
          return colors
docs: Simple function that adds fixed colors depending on the class