使用pytorch进行图像分割

Question

I am using pytorch for image segmentation. I want to calculate the masked area in the segmented picture below. My question is it it possible to calculate the masked area? If so how can i do that.

enter image description here

used torchvision. I want to know if there is any way to calculate the masked area of the boat in the segmented picture?

Answer 1

Let say you have a predicition output of the same shape as the input of shape [height, width] of type torch.long or torch.int where each element is the class index of its corresponding pixel:

seg_output = ...
BOAT_INDEX = ...

boat_area = torch.sum(seg_output == BOAT_INDEX)
boat_area_propotion = boat_area / (height * width)