builder.py 178 Bytes
Newer Older
unknown's avatar
unknown committed
1
2
3
4
5
6
7
from mmcv.utils import Registry, build_from_cfg

AUGMENT = Registry('augment')


def build_augment(cfg, default_args=None):
    return build_from_cfg(cfg, AUGMENT, default_args)