34 lines
1.8 KiB
Python
34 lines
1.8 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
# yapf: disable
|
|
from .class_names import (ade_classes, ade_palette, bdd100k_classes,
|
|
bdd100k_palette, cityscapes_classes,
|
|
cityscapes_palette, cocostuff_classes,
|
|
cocostuff_palette, dataset_aliases, get_classes,
|
|
get_palette, isaid_classes, isaid_palette,
|
|
loveda_classes, loveda_palette, potsdam_classes,
|
|
potsdam_palette, stare_classes, stare_palette,
|
|
synapse_classes, synapse_palette, vaihingen_classes,
|
|
vaihingen_palette, voc_classes, voc_palette)
|
|
# yapf: enable
|
|
from .collect_env import collect_env
|
|
from .io import datafrombytes
|
|
from .misc import add_prefix, stack_batch
|
|
from .set_env import register_all_modules
|
|
from .typing_utils import (ConfigType, ForwardResults, MultiConfig,
|
|
OptConfigType, OptMultiConfig, OptSampleList,
|
|
SampleList, TensorDict, TensorList)
|
|
|
|
__all__ = [
|
|
'collect_env', 'register_all_modules', 'stack_batch', 'add_prefix',
|
|
'ConfigType', 'OptConfigType', 'MultiConfig', 'OptMultiConfig',
|
|
'SampleList', 'OptSampleList', 'TensorDict', 'TensorList',
|
|
'ForwardResults', 'cityscapes_classes', 'ade_classes', 'voc_classes',
|
|
'cocostuff_classes', 'loveda_classes', 'potsdam_classes',
|
|
'vaihingen_classes', 'isaid_classes', 'stare_classes',
|
|
'cityscapes_palette', 'ade_palette', 'voc_palette', 'cocostuff_palette',
|
|
'loveda_palette', 'potsdam_palette', 'vaihingen_palette', 'isaid_palette',
|
|
'stare_palette', 'dataset_aliases', 'get_classes', 'get_palette',
|
|
'datafrombytes', 'synapse_palette', 'synapse_classes', 'bdd100k_classes',
|
|
'bdd100k_palette'
|
|
]
|