VainF / DeepLabV3Plus-Pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

classes colors

WeichengSong opened this issue · comments

Where do you modify the code of display colors for each category when forecasting?

Hi @WeichengSong, the colorization is done by dataset.decode_targets.

If you want to colorize your own datasets, you can use the VOC colormap. Here is an example of colorizing NYUv2 dataset. And a similar example can be found here in this repo.