chunk_content: d a color palette for this. Here, we use ade_palette(). As it is a long array, so
  we don’t include it in this guide, please copy it from the TensorFlow Model Garden repository.
  
  
  	Copied
  import matplotlib.pyplot as plt
  
  color_seg = np.zeros((pred_seg.shape[0], pred_seg.shape[1], 3), dtype=np.uint8)
  palette = np.array(ade_palette())
  
  for label, color in enumerate(palette):
      color_seg[pred_seg == label, :] = color
  color_seg = color_seg[..., ::