Function getColormap

deeplab/src/utils.ts:86–98

(base: ModelArchitecture)

 * to labels.
 */
export function getColormap(base: ModelArchitecture): Color[] {
  if (base === 'pascal') {
    return config['COLORMAPS']['PASCAL'] as Color[];
  } else if (base === 'ade20k') {
    return config['COLORMAPS']['ADE20K'] as Color[];
  } else if (base === 'cityscapes') {
    return config['COLORMAPS']['CITYSCAPES'] as Color[];
  }
  throw new Error(
      `SemanticSegmentation cannot be constructed ` +
      `with an invalid base model ${base}. ` +
      `Try one of 'pascal', 'cityscapes' and 'ade20k'.`);
}

/**
 * @param base  :: `ModelArchitecture`