mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
10 lines
332 B
Python
10 lines
332 B
Python
# Borrowed from https://github.com/open-mmlab/mmsegmentation/blob/master/mmseg
|
|
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)
|