pytorch-segmentation-toolbox icon indicating copy to clipboard operation
pytorch-segmentation-toolbox copied to clipboard

Published 20 hours ago • verified publisher icon speedinghzl

The visualization of the segmentation results.

Open chenyzh28 opened this issue 5 years ago • 5 comments

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. image. But the ground-truth label is like: image So how can i modify the program to make it right?

chenyzh28 avatar Sep 12 '19 05:09 chenyzh28

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

erichhhhho avatar Sep 12 '19 05:09 erichhhhho

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

chenyzh28 avatar Sep 12 '19 05:09 chenyzh28

Refer to here Column Color

erichhhhho avatar Sep 12 '19 06:09 erichhhhho

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

chenyzh28 avatar Sep 12 '19 06:09 chenyzh28

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

swjtulinxi avatar Dec 25 '20 09:12 swjtulinxi