From 2056caa790d9abd134ecf9eaa813b88121597a0e Mon Sep 17 00:00:00 2001 From: MengzhangLI Date: Wed, 16 Feb 2022 22:41:39 +0800 Subject: [PATCH] [Fix] Fix wrong palette value in vaihingen (#1292) --- mmseg/core/evaluation/class_names.py | 2 +- mmseg/datasets/isprs.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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):