Function getLabels

deeplab/src/utils.ts:109–121

(base: ModelArchitecture)

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

/**
 * @param input  ::