| |
| |
| 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) |
|
|
| |
| 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) |
|
|
| |
| 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', |
| ] |
|
|
