The visualization of the segmentation results.

[chenyzh28]

I trained the model with 4 GPUs (GTX 2080) with batchsize 4, and the PSPNet achieves the mIOU of 76.42%. But I found the visualization results are not correct. Here is an example.
.
But the ground-truth label is like:

So how can i modify the program to make it right?

[erichhhhho]

@chenyzh28 you can use the cityscapes colormap, this code uses a different color map. see get_palette()

[chenyzh28]

Thanks for your attention. But Where can I find the cityscapes colormap? Here is the original code of get_palette().

[erichhhhho]

Refer to here
Column Color

[chenyzh28]

Thank you very much! I have already corrected the colormap.

[swjtulinxi]

hi,i have a question about why your results are just 76.42%,you have 4 GPU,but your results are not the same as paper 79 or 78.x