Hey. I鈥檓 doing binary segmentation using Mask2FormerForUniversalSegmentation.from_pretrained and Mask2FormerImageProcessor. I鈥檓 following this tutorial Transformers-Tutorials/MaskFormer/Fine-tuning/Fine_tuning_MaskFormerForInstanceSegmentation_on_semantic_sidewalk.ipynb at master 路 NielsRogge/Transformers-Tutorials 路 GitHub
My custom dataset contains a single class. I鈥檓 not quite sure if Mask2Former supports binary segmentation or should I treat this task as a 2-class segmentation, considering 0-background and 1-myClass.
I have tried both cases but I鈥檓 struggling with the arguments ignore_index and reduce_labels.
The model does not learn and the IoU is around 0.2 for both classes (background and myClass) along all epochs.
I鈥檓 stuck in this point, any idea?
Thanks.