Decode batch of segmentation masks. Args: mask: result of inference after taking argmax. num_images: number of images to decode from the batch. Returns: A batch with num_images RGB images of the same size as the input.
(mask, num_images=1, num_classes=20)