problem in getting the output in Unet for more than 3 classes

Question

conv9 = Conv2D(2, 3, activation = \'relu\', padding = \'same\', kernel_initializer = he_normal\')(conv9)
conv10 = Conv2D(3, 1, activation = \'softmax\')(conv9)