How to get inference mask image , in which RGB values are correlating to cityscapes original label ids?

hellochick / PSPNet-tensorflow

TensorFlow-based implementation of "Pyramid Scene Parsing Network".

326 stars 123 forks source link

How to get inference mask image , in which RGB values are correlating to cityscapes original label ids? #64

Open LbinB opened 5 years ago

LbinB commented 5 years ago

I have added seg_map= sess.run(raw_output_up) in inference.py line112 and get mask images. However it's not related to cityscapes original label ids. Could you tell me how can i get inference mask image, in which RGB values are correlating to cityscapes original label ids? @hellochick thank you a lot