| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748 |
- # -------------------------------------------------------------------------
- # Copyright (c) 2021-2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
- #
- # NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual
- # property and proprietary rights in and to this software, related
- # documentation and any modifications thereto. Any use, reproduction,
- # disclosure or distribution of this software and related documentation
- # without an express license agreement from NVIDIA CORPORATION is strictly
- # prohibited.
- #
- # Written by Jiarui Xu
- # -------------------------------------------------------------------------
- # Modified by Jilan Xu
- # -------------------------------------------------------------------------
- from mmseg.datasets import DATASETS
- from mmseg.datasets import ADE20KDataset as _ADE20KDataset
- @DATASETS.register_module(force=True)
- class ADE20KDataset(_ADE20KDataset):
- CLASSES = (
- 'wall', 'building', 'sky', 'floor', 'tree', 'ceiling', 'road', 'bed ',
- 'windowpane', 'grass', 'cabinet', 'sidewalk', 'person', 'earth',
- 'door', 'table', 'mountain', 'plant', 'curtain', 'chair', 'car',
- 'water', 'painting', 'sofa', 'shelf', 'house', 'sea', 'mirror', 'rug',
- 'field', 'armchair', 'seat', 'fence', 'desk', 'rock', 'wardrobe',
- 'lamp', 'bathtub', 'railing', 'cushion', 'base', 'box', 'column',
- 'signboard', 'chest of drawers', 'counter', 'sand', 'sink',
- 'skyscraper', 'fireplace', 'refrigerator', 'grandstand', 'path',
- 'stairs', 'runway', 'case', 'pool table', 'pillow', 'screen door',
- 'stairway', 'river', 'bridge', 'bookcase', 'blind', 'coffee table',
- 'toilet', 'flower', 'book', 'hill', 'bench', 'countertop', 'stove',
- 'palm', 'kitchen island', 'computer', 'swivel chair', 'boat', 'bar',
- 'arcade machine', 'hovel', 'bus', 'towel', 'light', 'truck', 'tower',
- 'chandelier', 'awning', 'streetlight', 'booth', 'television receiver',
- 'airplane', 'dirt track', 'apparel', 'pole', 'land', 'bannister',
- 'escalator', 'ottoman', 'bottle', 'buffet', 'poster', 'stage', 'van',
- 'ship', 'fountain', 'conveyer belt', 'canopy', 'washer', 'plaything',
- 'swimming pool', 'stool', 'barrel', 'basket', 'waterfall', 'tent',
- 'bag', 'minibike', 'cradle', 'oven', 'ball', 'food', 'step', 'tank',
- 'trade name', 'microwave', 'pot', 'animal', 'bicycle', 'lake',
- 'dishwasher', 'screen', 'blanket', 'sculpture', 'hood', 'sconce',
- 'vase', 'traffic light', 'tray', 'ashcan', 'fan', 'pier', 'crt screen',
- 'plate', 'monitor', 'bulletin board', 'shower', 'radiator', 'glass',
- 'clock', 'flag')
-
|
