InternImage icon indicating copy to clipboard operation
InternImage copied to clipboard

Mask2Former config file error

Open ECNULchase opened this issue 2 years ago • 1 comments

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)

ECNULchase avatar Jul 26 '23 08:07 ECNULchase

https://github.com/OpenGVLab/InternImage/issues/219

DuinoDu avatar Aug 10 '23 08:08 DuinoDu