mirror of
https://github.com/open-mmlab/mmsegmentation.git
synced 2025-06-03 22:03:48 +08:00
[Fix] Add COCOStuffDataset in class_names.py. (#1222)
* add cocostuff in class_names * add more class names
This commit is contained in:
MengzhangLI
GitHub
parent
9afb7e1974
commit
bdecbdbdf1
@ -68,6 +68,41 @@ def potsdam_classes():
|
||||
]
|
||||
|
||||
|
||||
def cocostuff_classes():
|
||||
"""CocoStuff class names for external use."""
|
||||
return [
|
||||
'person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train',
|
||||
'truck', 'boat', 'traffic light', 'fire hydrant', 'stop sign',
|
||||
'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep',
|
||||
'cow', 'elephant', 'bear', 'zebra', 'giraffe', 'backpack', 'umbrella',
|
||||
'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard',
|
||||
'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard',
|
||||
'surfboard', 'tennis racket', 'bottle', 'wine glass', 'cup', 'fork',
|
||||
'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange',
|
||||
'broccoli', 'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair',
|
||||
'couch', 'potted plant', 'bed', 'dining table', 'toilet', 'tv',
|
||||
'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave',
|
||||
'oven', 'toaster', 'sink', 'refrigerator', 'book', 'clock', 'vase',
|
||||
'scissors', 'teddy bear', 'hair drier', 'toothbrush', 'banner',
|
||||
'blanket', 'branch', 'bridge', 'building-other', 'bush', 'cabinet',
|
||||
'cage', 'cardboard', 'carpet', 'ceiling-other', 'ceiling-tile',
|
||||
'cloth', 'clothes', 'clouds', 'counter', 'cupboard', 'curtain',
|
||||
'desk-stuff', 'dirt', 'door-stuff', 'fence', 'floor-marble',
|
||||
'floor-other', 'floor-stone', 'floor-tile', 'floor-wood', 'flower',
|
||||
'fog', 'food-other', 'fruit', 'furniture-other', 'grass', 'gravel',
|
||||
'ground-other', 'hill', 'house', 'leaves', 'light', 'mat', 'metal',
|
||||
'mirror-stuff', 'moss', 'mountain', 'mud', 'napkin', 'net', 'paper',
|
||||
'pavement', 'pillow', 'plant-other', 'plastic', 'platform',
|
||||
'playingfield', 'railing', 'railroad', 'river', 'road', 'rock', 'roof',
|
||||
'rug', 'salad', 'sand', 'sea', 'shelf', 'sky-other', 'skyscraper',
|
||||
'snow', 'solid-other', 'stairs', 'stone', 'straw', 'structural-other',
|
||||
'table', 'tent', 'textile-other', 'towel', 'tree', 'vegetable',
|
||||
'wall-brick', 'wall-concrete', 'wall-other', 'wall-panel',
|
||||
'wall-stone', 'wall-tile', 'wall-wood', 'water-other', 'waterdrops',
|
||||
'window-blind', 'window-other', 'wood'
|
||||
]
|
||||
|
||||
|
||||
def cityscapes_palette():
|
||||
"""Cityscapes palette for external use."""
|
||||
return [[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156],
|
||||
@ -140,12 +175,64 @@ def potsdam_palette():
|
||||
[255, 255, 0], [255, 0, 0]]
|
||||
|
||||
|
||||
def cocostuff_palette():
|
||||
"""CocoStuff palette for external use."""
|
||||
return [[0, 192, 64], [0, 192, 64], [0, 64, 96], [128, 192, 192],
|
||||
[0, 64, 64], [0, 192, 224], [0, 192, 192], [128, 192, 64],
|
||||
[0, 192, 96], [128, 192, 64], [128, 32, 192], [0, 0, 224],
|
||||
[0, 0, 64], [0, 160, 192], [128, 0, 96], [128, 0, 192],
|
||||
[0, 32, 192], [128, 128, 224], [0, 0, 192], [128, 160, 192],
|
||||
[128, 128, 0], [128, 0, 32], [128, 32, 0], [128, 0, 128],
|
||||
[64, 128, 32], [0, 160, 0], [0, 0, 0], [192, 128, 160], [0, 32, 0],
|
||||
[0, 128, 128], [64, 128, 160], [128, 160, 0], [0, 128, 0],
|
||||
[192, 128, 32], [128, 96, 128], [0, 0, 128], [64, 0, 32],
|
||||
[0, 224, 128], [128, 0, 0], [192, 0, 160], [0, 96, 128],
|
||||
[128, 128, 128], [64, 0, 160], [128, 224, 128], [128, 128, 64],
|
||||
[192, 0, 32], [128, 96, 0], [128, 0, 192], [0, 128, 32],
|
||||
[64, 224, 0], [0, 0, 64], [128, 128, 160], [64, 96, 0],
|
||||
[0, 128, 192], [0, 128, 160], [192, 224, 0], [0, 128, 64],
|
||||
[128, 128, 32], [192, 32, 128], [0, 64, 192], [0, 0, 32],
|
||||
[64, 160, 128], [128, 64, 64], [128, 0, 160], [64, 32, 128],
|
||||
[128, 192, 192], [0, 0, 160], [192, 160, 128], [128, 192, 0],
|
||||
[128, 0, 96], [192, 32, 0], [128, 64, 128], [64, 128, 96],
|
||||
[64, 160, 0], [0, 64, 0], [192, 128, 224], [64, 32, 0],
|
||||
[0, 192, 128], [64, 128, 224], [192, 160, 0], [0, 192, 0],
|
||||
[192, 128, 96], [192, 96, 128], [0, 64, 128], [64, 0, 96],
|
||||
[64, 224, 128], [128, 64, 0], [192, 0, 224], [64, 96, 128],
|
||||
[128, 192, 128], [64, 0, 224], [192, 224, 128], [128, 192, 64],
|
||||
[192, 0, 96], [192, 96, 0], [128, 64, 192], [0, 128, 96],
|
||||
[0, 224, 0], [64, 64, 64], [128, 128, 224], [0, 96, 0],
|
||||
[64, 192, 192], [0, 128, 224], [128, 224, 0], [64, 192, 64],
|
||||
[128, 128, 96], [128, 32, 128], [64, 0, 192], [0, 64, 96],
|
||||
[0, 160, 128], [192, 0, 64], [128, 64, 224], [0, 32, 128],
|
||||
[192, 128, 192], [0, 64, 224], [128, 160, 128], [192, 128, 0],
|
||||
[128, 64, 32], [128, 32, 64], [192, 0, 128], [64, 192, 32],
|
||||
[0, 160, 64], [64, 0, 0], [192, 192, 160], [0, 32, 64],
|
||||
[64, 128, 128], [64, 192, 160], [128, 160, 64], [64, 128, 0],
|
||||
[192, 192, 32], [128, 96, 192], [64, 0, 128], [64, 64, 32],
|
||||
[0, 224, 192], [192, 0, 0], [192, 64, 160], [0, 96, 192],
|
||||
[192, 128, 128], [64, 64, 160], [128, 224, 192], [192, 128, 64],
|
||||
[192, 64, 32], [128, 96, 64], [192, 0, 192], [0, 192, 32],
|
||||
[64, 224, 64], [64, 0, 64], [128, 192, 160], [64, 96, 64],
|
||||
[64, 128, 192], [0, 192, 160], [192, 224, 64], [64, 128, 64],
|
||||
[128, 192, 32], [192, 32, 192], [64, 64, 192], [0, 64, 32],
|
||||
[64, 160, 192], [192, 64, 64], [128, 64, 160], [64, 32, 192],
|
||||
[192, 192, 192], [0, 64, 160], [192, 160, 192], [192, 192, 0],
|
||||
[128, 64, 96], [192, 32, 64], [192, 64, 128], [64, 192, 96],
|
||||
[64, 160, 64], [64, 64, 0]]
|
||||
|
||||
|
||||
dataset_aliases = {
|
||||
'cityscapes': ['cityscapes'],
|
||||
'ade': ['ade', 'ade20k'],
|
||||
'voc': ['voc', 'pascal_voc', 'voc12', 'voc12aug'],
|
||||
'loveda': ['loveda'],
|
||||
'potsdam': ['potsdam']
|
||||
'potsdam': ['potsdam'],
|
||||
'cocostuff': [
|
||||
'cocostuff', 'cocostuff10k', 'cocostuff164k', 'coco-stuff',
|
||||
'coco-stuff10k', 'coco-stuff164k', 'coco_stuff', 'coco_stuff10k',
|
||||
'coco_stuff164k'
|
||||
]
|
||||
}
|
||||
|
||||
|
||||
|
||||
@ -13,9 +13,10 @@ from PIL import Image
|
||||
|
||||
from mmseg.core.evaluation import get_classes, get_palette
|
||||
from mmseg.datasets import (DATASETS, ADE20KDataset, CityscapesDataset,
|
||||
ConcatDataset, CustomDataset, LoveDADataset,
|
||||
MultiImageMixDataset, PascalVOCDataset,
|
||||
PotsdamDataset, RepeatDataset, build_dataset)
|
||||
COCOStuffDataset, ConcatDataset, CustomDataset,
|
||||
LoveDADataset, MultiImageMixDataset,
|
||||
PascalVOCDataset, PotsdamDataset, RepeatDataset,
|
||||
build_dataset)
|
||||
|
||||
|
||||
def test_classes():
|
||||
@ -26,6 +27,7 @@ def test_classes():
|
||||
ADE20KDataset.CLASSES) == get_classes('ade') == get_classes('ade20k')
|
||||
assert list(LoveDADataset.CLASSES) == get_classes('loveda')
|
||||
assert list(PotsdamDataset.CLASSES) == get_classes('potsdam')
|
||||
assert list(COCOStuffDataset.CLASSES) == get_classes('cocostuff')
|
||||
|
||||
with pytest.raises(ValueError):
|
||||
get_classes('unsupported')
|
||||
@ -69,6 +71,7 @@ def test_palette():
|
||||
assert ADE20KDataset.PALETTE == get_palette('ade') == get_palette('ade20k')
|
||||
assert LoveDADataset.PALETTE == get_palette('loveda')
|
||||
assert PotsdamDataset.PALETTE == get_palette('potsdam')
|
||||
assert COCOStuffDataset.PALETTE == get_palette('cocostuff')
|
||||
|
||||
with pytest.raises(ValueError):
|
||||
get_palette('unsupported')
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user