Spaces:
Runtime error
Runtime error
# Copyright (c) OpenMMLab. All rights reserved. | |
from mmdet.registry import MODELS | |
from mmdet.utils import ConfigType, OptConfigType, OptMultiConfig | |
from .single_stage_instance_seg import SingleStageInstanceSegmentor | |
class SOLOv2(SingleStageInstanceSegmentor): | |
"""`SOLOv2: Dynamic and Fast Instance Segmentation | |
<https://arxiv.org/abs/2003.10152>`_ | |
""" | |
def __init__(self, | |
backbone: ConfigType, | |
neck: OptConfigType = None, | |
bbox_head: OptConfigType = None, | |
mask_head: OptConfigType = None, | |
train_cfg: OptConfigType = None, | |
test_cfg: OptConfigType = None, | |
data_preprocessor: OptConfigType = None, | |
init_cfg: OptMultiConfig = None): | |
super().__init__( | |
backbone=backbone, | |
neck=neck, | |
bbox_head=bbox_head, | |
mask_head=mask_head, | |
train_cfg=train_cfg, | |
test_cfg=test_cfg, | |
data_preprocessor=data_preprocessor, | |
init_cfg=init_cfg) | |