Python Registryの例

コード例 #1

ファイル: registry.py プロジェクト: AHP-boop/Ped-detection

from mmdet.utils import Registry

DATASETS = Registry('dataset')

コード例 #2

ファイル: registry.py プロジェクト: AHP-boop/Ped-detection

from mmdet.utils import Registry

BACKBONES = Registry('backbone')
NECKS = Registry('neck')
ROI_EXTRACTORS = Registry('roi_extractor')
SHARED_HEADS = Registry('shared_head')
HEADS = Registry('head')
LOSSES = Registry('loss')
DETECTORS = Registry('detector')

コード例 #3

ファイル: registry.py プロジェクト: zyg11/TSD

from mmdet.utils import Registry

DATASETS = Registry("dataset")
PIPELINES = Registry("pipeline")

コード例 #4

import inspect

import torch

from mmdet.utils import Registry

OPTIMIZERS = Registry("optimizer")


def register_torch_optimizers():
    torch_optimizers = []
    for module_name in dir(torch.optim):
        if module_name.startswith("__"):
            continue
        _optim = getattr(torch.optim, module_name)
        if inspect.isclass(_optim) and issubclass(_optim,
                                                  torch.optim.Optimizer):
            OPTIMIZERS.register_module(_optim)
            torch_optimizers.append(module_name)
    return torch_optimizers


TORCH_OPTIMIZERS = register_torch_optimizers()

コード例 #5

ファイル: registry.py プロジェクト: ktw361/mmdetection_impl

from mmdet.utils import Registry

BACKBONES = Registry('backbone')
NECKS = Registry('neck')
TEMPORAL_MODULE = Registry('temporal_module')
PAIR_MODULE = Registry('pair_module')
ROI_EXTRACTORS = Registry('roi_extractor')
SHARED_HEADS = Registry('shared_head')
HEADS = Registry('head')
LOSSES = Registry('loss')
DETECTORS = Registry('detector')

コード例 #6

ファイル: registry.py プロジェクト: ZDstandup/LIO

from mmdet.utils import Registry

DATASETS = Registry('dataset')
PIPELINES = Registry('pipeline')

コード例 #7

ファイル: registry.py プロジェクト: zyg11/TSD

from mmdet.utils import Registry

BACKBONES = Registry("backbone")
NECKS = Registry("neck")
ROI_EXTRACTORS = Registry("roi_extractor")
SHARED_HEADS = Registry("shared_head")
HEADS = Registry("head")
LOSSES = Registry("loss")
DETECTORS = Registry("detector")

コード例 #8

from mmdet.utils import Registry, build_from_cfg

BBOX_ASSIGNERS = Registry('bbox_assigner')
BBOX_SAMPLERS = Registry('bbox_sampler')
BBOX_CODERS = Registry('bbox_coder')


def build_assigner(cfg, **default_args):
    """Builder of box assigner."""
    return build_from_cfg(cfg, BBOX_ASSIGNERS, default_args)


def build_sampler(cfg, **default_args):
    """Builder of box sampler."""
    return build_from_cfg(cfg, BBOX_SAMPLERS, default_args)


def build_bbox_coder(cfg, **default_args):
    """Builder of box coder."""
    return build_from_cfg(cfg, BBOX_CODERS, default_args)

コード例 #9

from mmdet.utils import Registry

DATASETS = Registry('dataset')
PIPELINES = Registry('pipeline')  # the name is pipeline, use the builder.py

コード例 #10

ファイル: builder.py プロジェクト: Rooooyy/BUAA_PR

from mmdet.utils import Registry, build_from_cfg

IOU_CALCULATORS = Registry('IoU calculator')


def build_iou_calculator(cfg, default_args=None):
    """Builder of IoU calculator."""
    return build_from_cfg(cfg, IOU_CALCULATORS, default_args)

コード例 #11

ファイル: registry.py プロジェクト: yakovlevartem450/mmdetection-artem

import inspect

import torch

from mmdet.utils import Registry

OPTIMIZERS = Registry('optimizer')


def register_torch_optimizers():
    torch_optimizers = []
    for module_name in dir(torch.optim):
        if module_name.startswith('__'):
            continue
        _optim = getattr(torch.optim, module_name)
        if inspect.isclass(_optim) and issubclass(_optim,
                                                  torch.optim.Optimizer):
            OPTIMIZERS.register_module(_optim)
            torch_optimizers.append(module_name)
    return torch_optimizers


TORCH_OPTIMIZERS = register_torch_optimizers()