mmsegmentation/mmseg/core/seg/builder.py

9 lines
253 B
Python
Raw Normal View History

2020-07-07 20:52:19 +08:00
from mmcv.utils import Registry, build_from_cfg
PIXEL_SAMPLERS = Registry('pixel sampler')
def build_pixel_sampler(cfg, **default_args):
"""Build pixel sampler for segmentation map."""
return build_from_cfg(cfg, PIXEL_SAMPLERS, default_args)