示例#1
0
from mmdet.utils import Registry

DATASETS = Registry('dataset')
示例#2
0
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
0
文件： registry.py 项目： zyg11/TSD 
from mmdet.utils import Registry

DATASETS = Registry("dataset")
PIPELINES = Registry("pipeline")
示例#4
0
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
0
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
0
文件： registry.py 项目： ZDstandup/LIO 
from mmdet.utils import Registry

DATASETS = Registry('dataset')
PIPELINES = Registry('pipeline')
示例#7
0
文件： 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
0
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
0
from mmdet.utils import Registry

DATASETS = Registry('dataset')
PIPELINES = Registry('pipeline')  # the name is pipeline, use the builder.py
示例#10
0
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
0
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()