search

hellochick / PSPNet-tensorflow

TensorFlow-based implementation of "Pyramid Scene Parsing Network".
326 stars 123 forks source link

How to get (class) label IDs or train IDs for evaluation with cityscapes-scripts #60

Open ga84 opened 5 years ago

ga84 commented 5 years ago

If anyone has found a way to do this, could you please give me a hint? I guess there's simpler way than writing script to convert RGB segmentation output into these IDs (that could get complicated, since some classes have the same RGB -e.g. pole and pole group).

I tried to solve the problem with this code by getting IDs as: IDs = sess.run(raw_output_up) and then saving them, however this outputs an image with numbers which don't represent neither train IDs, nor class labels (numbers up to 255, but don't fit to these two IDs).... @jiaxinxie97 @Jason-xin @MichaelX99 @hellochick Thank you!

LbinB commented 5 years ago

Did you solve this problem? I met the same problem too