Image segmentation using 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.

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

what do you mean, "calculate"? the areas are right there. you don't calculate them. what more do you need? you could calculate specific information on those areas. you didn't say that though. -- welcome. tour, How to Ask, minimal reproducible example — Christoph Rackwitz
– Christoph Rackwitz, Commented Dec 19, 2022 at 2:56
In this image you can clearly see that the model segmented areas representing a boat, a person, and probably the background. To get the same result you need to train an image segmentation model where it can predict for each pixel whether it belongs to one of the known classes. — NotAName
– NotAName, Commented Dec 19, 2022 at 3:01
lets say the image is 255*255. Then what portion of the image is Boat . Is there any way i can count the pixels for the boat and determine the area of it compared to the total image. — AHR99
– AHR99, Commented Dec 19, 2022 at 4:08
this question looks similar to another question you ask in the future: stackoverflow.com/questions/74936584/… — Christoph Rackwitz
– Christoph Rackwitz, Commented Jan 4, 2023 at 2:12

cao-nv · Accepted Answer · 2022-12-19 04:25:35Z

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)

answered Dec 19, 2022 at 4:25

cao-nv

4043 silver badges6 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

AHR99 Over a year ago

could you please explain further on this. Thanks.

cao-nv Over a year ago

Just compare the output with the index of the "boat" categories, then sum the number of true predicted points to get the area.

AHR99 Over a year ago

Thank you for your generous comment. I am new to this. But I am unsure how to get the index. I am sharing the code link. Could you please help me on this? stackoverflow.com/questions/74936584/…

Collectives™ on Stack Overflow

Image segmentation using pytorch

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related