segformer_plusplus/utils/__init__.py

# Copyright (c) OpenMMLab. All rights reserved.
from .embed import PatchEmbed
from .shape_convert import nchw_to_nlc, nlc_to_nchw
from .wrappers import resize, Conv2d
from .tome_presets import tome_presets
from .registry import MODELS
from .imagenet_weights import imagenet_weights
from .benchmark import benchmark
from .activation import build_activation_layer, build_norm_layer, build_dropout
from .version_utils import digit_version


__all__ = [
    'PatchEmbed', 'nchw_to_nlc', 'nlc_to_nchw', 'resize', 'Conv2d', 'tome_presets', 'MODELS', 'imagenet_weights', 'benchmark', 'build_activation_layer', 'build_norm_layer', 'build_dropout', 'digit_version'
]