Takes in a matrix HxWx1 where each pixel is a label in the set {0,..,k} and returns a matrix of size HxWx(k+1)

SegmentationOneHotEncoder.py

"""
One-hot-encodes segmentation map
@params
y: np.ndarray
  HxWx1 seegmentation map where each element is in set {0,...,num_labels}
num_labels: int
  Defines the set {0,...,num_labels} which is used to one-hot-encode the segmentation map
  Ex. For annotation segmentation, we have 26 letters, so we say 26. It is assumed that a 0 means no class of interest.
@return
  target: np.ndarray
    HxWx(num_labels+1) segmentation matrix
"""
def convert_to_segmentation_ground_truth(y,num_labels=26):
  target = np.zeros((num_labels+1,128,128))
  target[0,:,:] = np.ones((128,128))
  for i in range(1,num_labels+1):
    seg_one = y  == i
    target[i,:,:] = seg_one
    target[0,:,:] = target[0,:,:] - target[i,:,:]
  target = target.astype(np.float32)
  return target