zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.
https://hangzhang.org/PyTorch-Encoding/
MIT License
2.04k stars 452 forks source link

question about generate mask #278

Open bychen7 opened 4 years ago

bychen7 commented 4 years ago

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?