Running on a data without test labels about segmentation_models HOT 4 CLOSED

Hi, @absagargupta
You can make prediction on any images/any dataset (make sure to preprocess them same as in training stage), but model will produce label it have been trained on.

from segmentation_models.

Thank you so much for the prompt reply. In case I need just the labels of people and the background what changes should I take care of?

from segmentation_models.

Suppose you have trained you model for C=10 classes, where value 3 is people class.
Output of network will be a tensor output with shape (N, H, W, 10).
To get mask of people class:
mask = (np.argmax(output, axis=-1) == 3)

from segmentation_models.

Thank you. :-)

from segmentation_models.