mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-23 19:49:24 +01:00
23 lines
805 B
Python
23 lines
805 B
Python
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||
|
|
|
||
|
|
from maas_lib.utils.config import ConfigDict
|
||
|
|
from maas_lib.utils.constant import Fields
|
||
|
|
from maas_lib.utils.registry import Registry, build_from_cfg
|
||
|
|
|
||
|
|
PREPROCESSORS = Registry('preprocessors')
|
||
|
|
|
||
|
|
|
||
|
|
def build_preprocessor(cfg: ConfigDict,
|
||
|
|
field_name: str = None,
|
||
|
|
default_args: dict = None):
|
||
|
|
""" build preprocesor given model config dict
|
||
|
|
|
||
|
|
Args:
|
||
|
|
cfg (:obj:`ConfigDict`): config dict for model object.
|
||
|
|
field_name (str, optional): application field name, refer to
|
||
|
|
:obj:`Fields` for more details
|
||
|
|
default_args (dict, optional): Default initialization arguments.
|
||
|
|
"""
|
||
|
|
return build_from_cfg(
|
||
|
|
cfg, PREPROCESSORS, group_key=field_name, default_args=default_args)
|