Segment multiple objects simultaneously on the same mask for images.

[sajeedmehrab]

Is there a way to segment multiple objects simultaneously on the same output mask for images?

Let's say I want to segment B number of objects in my image. Let's also assume that I have n number of foreground points for each of those B objects. If create a point_coords batch with the shape [B, n, 2], with multimask_output set to False, I get an output of shape [B, 1, img_height, img_width]. Essentially, I get B number of binary masks -- one for each of my B objects.

Is there a way of getting a single non-binary mask, such that each object will be masked with B unique object_id's or class labels?

I saw that this can be done by assigning object_id's to videos, but I was wondering if there's a way of doing this for images.