InternImage
InternImage copied to clipboard
Mask2Former config file error
In segmentation/configs/base/models/mask2former_beit.py
decode_head=dict(
type='Mask2FormerHead',
in_channels=[256, 512, 1024, 2048], # pass to pixel_decoder inside
# strides=[4, 8, 16, 32],
feat_channels=256,
out_channels=256,
in_index=[0, 1, 2, 3],
num_things_classes=num_things_classes,
num_stuff_classes=num_stuff_classes,
num_queries=100,
num_transformer_feat_level=3,
there are num_things_classes and num_stuff_classes but not in segmentation/mmseg_custom/models/decode_heads/mask2former_head.py
class Mask2FormerHead(BaseDecodeHead):
def __init__(self,
in_channels,
feat_channels,
out_channels,
num_classes=80,
num_queries=100,
num_transformer_feat_level=3,
pixel_decoder=None,
enforce_decoder_input_project=False,
transformer_decoder=None,
positional_encoding=None,
loss_cls=None,
loss_mask=None,
loss_dice=None,
train_cfg=None,
test_cfg=None,
init_cfg=None,
**kwargs):
super(Mask2FormerHead, self).__init__(
in_channels=in_channels,
channels=feat_channels,
num_classes=num_classes,
init_cfg=init_cfg,
input_transform='multiple_select',
**kwargs)
https://github.com/OpenGVLab/InternImage/issues/219