diff --git a/mmseg/core/evaluation/class_names.py b/mmseg/core/evaluation/class_names.py index cc90517b4..65d8a49e8 100644 --- a/mmseg/core/evaluation/class_names.py +++ b/mmseg/core/evaluation/class_names.py @@ -232,7 +232,7 @@ def potsdam_palette(): def vaihingen_palette(): """Vaihingen palette for external use.""" - return [[255, 255, 255], [0, 0, 255], [0, 0, 255], [0, 255, 0], + return [[255, 255, 255], [0, 0, 255], [0, 255, 255], [0, 255, 0], [255, 255, 0], [255, 0, 0]] diff --git a/mmseg/datasets/isprs.py b/mmseg/datasets/isprs.py index 504a02233..5f23e1a9b 100644 --- a/mmseg/datasets/isprs.py +++ b/mmseg/datasets/isprs.py @@ -14,7 +14,7 @@ class ISPRSDataset(CustomDataset): CLASSES = ('impervious_surface', 'building', 'low_vegetation', 'tree', 'car', 'clutter') - PALETTE = [[255, 255, 255], [0, 0, 255], [0, 0, 255], [0, 255, 0], + PALETTE = [[255, 255, 255], [0, 0, 255], [0, 255, 255], [0, 255, 0], [255, 255, 0], [255, 0, 0]] def __init__(self, **kwargs):