How can we get the area of each segment in Semantic segmentation.

[manika1511]

I am trying to get the area (number of pixels) of each segment using the Semantic segmentation result. Based on the model output format here https://detectron2.readthedocs.io/en/latest/tutorials/models.html#model-output-format, "sem_seg" output is num_categoriesxHxW.