code: def colormap(rgb=False, maximum=255):
      """
      Args:
          rgb (bool): whether to return RGB colors or BGR colors.
          maximum (int): either 255 or 1
      Returns:
          ndarray: a float32 array of Nx3 colors, in range [0, 255] or [0, 1]
      """
      assert maximum in [255, 1], maximum
      c = _COLORS * maximum
      if not rgb:
          c = c[:, ::-1]
      return c
  
imports: import colorsys
  import logging
  import math
  from enum import Enum, unique
  import numpy
  import matplotlib
  import matplotlib.colors
  import matplotlib.figure
  import torch
  from matplotlib.backends.backend_agg import FigureCanvasAgg
  from PIL import Image
  from ..data.registry.catalog import MetadataCatalog
  from ..data.structures import BitMasks, Boxes, BoxMode, Keypoints, PolygonMasks
label_desc: Use image node_id 2 for calling a global function with example usage: colormap(rgb, maximum) and returns: c