You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
128 lines
3.5 KiB
Python
128 lines
3.5 KiB
Python
# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from .registry import BACKBONES, HEADS, LOSSES, RECOGNIZERS, LOCALIZERS, ROI_EXTRACTORS, DETECTORS, BBOX_ASSIGNERS, BBOX_SAMPLERS, BBOX_CODERS, PARTITIONERS, MULTIMODAL, SEGMENT, SEGMENTERS
|
|
from ..utils import build
|
|
from .registry import (BACKBONES, BBOX_ASSIGNERS, BBOX_CODERS, BBOX_SAMPLERS,
|
|
DETECTORS, ESTIMATORS, HEADS, LOCALIZERS, LOSSES,
|
|
MULTIMODAL, PARTITIONERS, RECOGNIZERS, ROI_EXTRACTORS)
|
|
|
|
|
|
def build_backbone(cfg):
|
|
"""Build backbone."""
|
|
return build(cfg, BACKBONES)
|
|
|
|
|
|
def build_roi_extractor(cfg):
|
|
"""Build roi extractor."""
|
|
return build(cfg, ROI_EXTRACTORS)
|
|
|
|
|
|
def build_assigner(cfg, **default_args):
|
|
"""Builder of box assigner."""
|
|
return build(cfg, BBOX_ASSIGNERS)
|
|
|
|
|
|
def build_sampler(cfg, **default_args):
|
|
"""Builder of box sampler."""
|
|
return build(cfg, BBOX_SAMPLERS)
|
|
|
|
|
|
def build_roi_extractor(cfg):
|
|
"""Build roi extractor."""
|
|
return build(cfg, ROI_EXTRACTORS)
|
|
|
|
|
|
def build_assigner(cfg, **default_args):
|
|
"""Builder of box assigner."""
|
|
return build(cfg, BBOX_ASSIGNERS)
|
|
|
|
|
|
def build_sampler(cfg, **default_args):
|
|
"""Builder of box sampler."""
|
|
return build(cfg, BBOX_SAMPLERS)
|
|
|
|
|
|
def build_head(cfg):
|
|
"""Build head."""
|
|
return build(cfg, HEADS)
|
|
|
|
|
|
def build_loss(cfg):
|
|
"""Build loss."""
|
|
return build(cfg, LOSSES)
|
|
|
|
|
|
def build_recognizer(cfg):
|
|
"""Build recognizer."""
|
|
return build(cfg, RECOGNIZERS, key='framework')
|
|
|
|
|
|
def build_segmenter(cfg):
|
|
"""Build segmenter."""
|
|
return build(cfg, SEGMENTERS, key='framework')
|
|
|
|
|
|
def build_localizer(cfg):
|
|
"""Build localizer."""
|
|
return build(cfg, LOCALIZERS, key='framework')
|
|
|
|
|
|
def build_detector(cfg, train_cfg=None, test_cfg=None):
|
|
"""Build detector."""
|
|
return build(cfg, DETECTORS, key='framework')
|
|
|
|
|
|
def build_partitioner(cfg):
|
|
"""Build partitioner."""
|
|
return build(cfg, PARTITIONERS, key='framework')
|
|
|
|
|
|
def build_estimator(cfg):
|
|
"""Build estimator."""
|
|
return build(cfg, ESTIMATORS, key='framework')
|
|
|
|
|
|
def build_multimodal(cfg):
|
|
"""Build multimodal."""
|
|
return build(cfg, MULTIMODAL, key='framework')
|
|
|
|
|
|
def build_segment(cfg):
|
|
"""Build segment."""
|
|
return build(cfg, SEGMENT, key='framework')
|
|
|
|
|
|
def build_model(cfg):
|
|
cfg_copy = cfg.copy()
|
|
framework_type = cfg_copy.get('framework')
|
|
if framework_type in RECOGNIZERS:
|
|
return build_recognizer(cfg)
|
|
elif framework_type in LOCALIZERS:
|
|
return build_localizer(cfg)
|
|
elif framework_type in PARTITIONERS:
|
|
return build_partitioner(cfg)
|
|
elif framework_type in DETECTORS:
|
|
return build_detector(cfg)
|
|
elif framework_type in ESTIMATORS:
|
|
return build_estimator(cfg)
|
|
elif framework_type in MULTIMODAL:
|
|
return build_multimodal(cfg)
|
|
elif framework_type in SEGMENTERS:
|
|
return build_segmenter(cfg)
|
|
elif framework_type in SEGMENT:
|
|
return build_segment(cfg)
|
|
else:
|
|
raise NotImplementedError
|