Force number of masks for the panoptic segmentation

facebookresearch / Mask2Former

Code release for "Masked-attention Mask Transformer for Universal Image Segmentation"

MIT License

2.59k stars 388 forks source link

Force number of masks for the panoptic segmentation #193

Open guist opened 1 year ago

guist commented 1 year ago

Hello,

I fine-tuned the panoptic segmentation model with my custom dataset, where I have two classes (background and object).

I wanted to ask if it is possible to force the number of object masks to be detected on an image? For example, if I know that my image contains 3 objects, I would like to get 1 mask for the background and 3 masks for the objects.

Thanks!