about discrete_matshow not same as tutorial

[LemonAniLabs]

Hi, when I try to run your tutorial to make a segmentation with vgg16, I got the result, it's look well.
But it's a little different with your result image.

This is my result.

I'd like to tune the color map(or color bar) as yours.
But I don't know how to do.

Thanks!!