Convert color to segmentation mask.
rgb (np.ndarray) – a rgb image
remap (dict) – like {index: (r,g,b)}
{index: (r,g,b)}
Python API