PyTorch-Encoding
PyTorch-Encoding copied to clipboard
question about generate mask
https://github.com/zhanghang1989/PyTorch-Encoding/blob/d9dea1724e38362a7c75ca9498f595248f283f00/encoding/datasets/pascal_voc.py#L24
In this folder,the image was color mask,but during training i saved mask image was class 1~N,and i did't find any code to transform the color mask to class 1~N,how to do that?