why predict values are strange

[liangdashuang]

I trained a model with my own data, the test output are as follows with "pred.py":

it seems there is something wrong, what's the reason?

I trained with my own data, my mask's pixel value is 0,1,2,3,4

[akirasosa]

@liangdashuang I'm not sure it's wrong or not. Have you tried to draw it?

[liangdashuang]

@akirasosa I only have 5 class(include the backgroud), the mask value is 0,1,2,3,4, so the predict output value should be 0,1,2,3,4 as well, am i right?

[akirasosa]

@liangdashuang
In this case, your output shape will be (h, w, 5). It means you have 5 channels.