How can I get the labels of the segment mask?

from keras.preprocessing import image
from segmentation_models import Unet
import numpy as np
import imageio

model = Unet(backbone_name='resnet152',
             encoder_weights='imagenet', freeze_encoder=True)
model.compile('Adam', 'binary_crossentropy', ['binary_accuracy'])

img = image.load_img('seg_images/image1.png')
x = image.img_to_array(img)
x = np.expand_dims(x, axis=0)

mask = model.predict(x)
mask = mask.astype('uint8')
imageio.imwrite("result.png", mask[0])

First, the mask is all black. But assuming it works, how can I get the various labels for each segment?

[qubvel]

Hi @shamoons If I understand you correctly: Round mask - mask.round() 0 - is background 1 - is target class

[shamoons]

How would I load it?