2021-08-17 14:16:55 +08:00
|
|
|
# Copyright (c) OpenMMLab. All rights reserved.
|
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)
|