halbielee / EPS

Official PyTorch implementation of "Railroad is not a Train: Saliency as Pseudo-pixel Supervision for Weakly Supervised Semantic Segmentation", CVPR2021
89 stars 13 forks source link

About the color map of coco 2014 #14

Closed XuYin94 closed 1 year ago

XuYin94 commented 2 years ago

Thanks for your wonderful work on WSSS.

Could you provide the color map (palette) in your work (on MS coco 2014) for segmentation visualization?

halbielee commented 1 year ago

Sorry for the late reply.

I just used the color map from PASCAL VOC 2012. If you use PIL module, when loading the color image from PASCAL VOC 2012, you can extract the color palette!

(see Image.getpalette() / Image.putpalette())