Open AndreiMihalea opened 2 months ago
Hello,
As I see in the file InternImage/segmentation/mmseg_custom/models/decode_heads /mask2former_head.py, there is this line:
InternImage/segmentation/mmseg_custom/models/decode_heads /mask2former_head.py
self.cls_embed = nn.Linear(feat_channels, self.num_classes + 1)
I am wondering why the +1 is there. As far as I've understood, it's for the background class. But what if my dataset already has a background class? Should I ignore it in the ToMask transform?
+1
ToMask
Hello,
As I see in the file
InternImage/segmentation/mmseg_custom/models/decode_heads /mask2former_head.py
, there is this line:self.cls_embed = nn.Linear(feat_channels, self.num_classes + 1)
I am wondering why the
+1
is there. As far as I've understood, it's for the background class. But what if my dataset already has a background class? Should I ignore it in theToMask
transform?