File size: 3,484 Bytes
6ed4a9c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 | # 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,
deepfashion_palette, deepfashion_classes,
fashion_3category_classes, fashion_3category_palette,
human_parsing_classes, human_parsing_palette,
lip_classes, lip_palette,
human_union_classes, human_union_palette,
deep_fashion_10k_classes, deep_fashion_10k_palette,
imaterialist_classes, imaterialist_palette,
imaterialist_5cat_classes, imaterialist_5cat_palette,
union_new_classes, union_new_palette,
union_new_add_mask_classes, union_new_add_mask_palette)
# yapf: enable
from .collect_env import collect_env
from .get_templates import get_predefined_templates
from .io import datafrombytes
from .misc import add_prefix, stack_batch
from .set_env import register_all_modules
from .tokenizer import tokenize
from .typing_utils import (ConfigType, ForwardResults, MultiConfig,
OptConfigType, OptMultiConfig, OptSampleList,
SampleList, TensorDict, TensorList)
# isort: off
from .mask_classification import MatchMasks, seg_data_to_instance_data
__all__ = [
'union_new_add_mask_classes',
'union_new_add_mask_palette',
'union_new_classes',
'union_new_palette',
'imaterialist_5cat_classes',
'imaterialist_5cat_palette',
'imaterialist_classes',
'imaterialist_palette',
'human_union_classes',
'human_union_palette',
'deep_fashion_10k_classes',
'deep_fashion_10k_palette',
'lip_classes',
'lip_palette',
'human_parsing_classes',
'human_parsing_palette',
'fashion_3category_palette',
'fashion_3category_classes',
'deepfashion_palette',
'deepfashion_classes',
'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',
'get_predefined_templates',
'tokenize',
'seg_data_to_instance_data',
'MatchMasks',
'bdd100k_classes',
'bdd100k_palette',
]
|