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.
30 lines
870 B
30 lines
870 B
# Copyright (c) OpenMMLab. All rights reserved. |
|
from ..builder import DETECTORS |
|
from .single_stage_instance_seg import SingleStageInstanceSegmentor |
|
|
|
|
|
@DETECTORS.register_module() |
|
class SOLO(SingleStageInstanceSegmentor): |
|
"""`SOLO: Segmenting Objects by Locations |
|
<https://arxiv.org/abs/1912.04488>`_ |
|
|
|
""" |
|
|
|
def __init__(self, |
|
backbone, |
|
neck=None, |
|
bbox_head=None, |
|
mask_head=None, |
|
train_cfg=None, |
|
test_cfg=None, |
|
init_cfg=None, |
|
pretrained=None): |
|
super().__init__( |
|
backbone=backbone, |
|
neck=neck, |
|
bbox_head=bbox_head, |
|
mask_head=mask_head, |
|
train_cfg=train_cfg, |
|
test_cfg=test_cfg, |
|
init_cfg=init_cfg, |
|
pretrained=pretrained)
|
|
|