OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
27 lines
840 B
27 lines
840 B
# Copyright (c) OpenMMLab. All rights reserved. |
|
from ..builder import DETECTORS |
|
from .maskformer import MaskFormer |
|
|
|
|
|
@DETECTORS.register_module() |
|
class Mask2Former(MaskFormer): |
|
r"""Implementation of `Masked-attention Mask |
|
Transformer for Universal Image Segmentation |
|
<https://arxiv.org/pdf/2112.01527>`_.""" |
|
|
|
def __init__(self, |
|
backbone, |
|
neck=None, |
|
panoptic_head=None, |
|
panoptic_fusion_head=None, |
|
train_cfg=None, |
|
test_cfg=None, |
|
init_cfg=None): |
|
super().__init__( |
|
backbone, |
|
neck=neck, |
|
panoptic_head=panoptic_head, |
|
panoptic_fusion_head=panoptic_fusion_head, |
|
train_cfg=train_cfg, |
|
test_cfg=test_cfg, |
|
init_cfg=init_cfg)
|
|
|