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