EasyCV/easycv/utils/__init__.py
2022-04-02 20:01:06 +08:00

19 lines
886 B
Python

# Copyright (c) Alibaba, Inc. and its affiliates.
"""
isort:skip_file
"""
from .alias_multinomial import AliasMethod
from .checkpoint import load_checkpoint
from .collect import dist_forward_collect, nondist_forward_collect
from .collect_env import collect_env
from .config_tools import mmcv_config_fromfile, traverse_replace
from .eval_utils import generate_best_metric_name
from .flops_counter import get_model_complexity_info, get_model_info
from .logger import get_root_logger, print_log
from .metric_distance import CosineSimilarity, DotproductSimilarity, LpDistance
from .preprocess_function import (bninceptionPre, gaussianBlur,
gaussianBlurDynamic, mixUp, mixUpCls,
randomErasing, randomGrayScale, solarize)
from .registry import Registry, build_from_cfg
from .user_config_params_utils import check_value_type