From ce3d954a00025bdbec6a38cee295f71ad7006105 Mon Sep 17 00:00:00 2001 From: ccoding Date: Tue, 15 Mar 2022 15:25:57 +0800 Subject: [PATCH 1/2] Support #1375: add demo/image_demo.py support for STARE --- mmseg/core/evaluation/class_names.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/mmseg/core/evaluation/class_names.py b/mmseg/core/evaluation/class_names.py index 447ed34bfb..f900e3ed3a 100644 --- a/mmseg/core/evaluation/class_names.py +++ b/mmseg/core/evaluation/class_names.py @@ -120,6 +120,9 @@ def isaid_classes(): 'Soccer_ball_field', 'plane', 'Harbor' ] +def stare_classes(): + """stare class names for external use.""" + return ['background', 'vessel'] def cityscapes_palette(): """Cityscapes palette for external use.""" @@ -254,6 +257,9 @@ def isaid_palette(): [0, 0, 191], [0, 0, 255], [0, 191, 127], [0, 127, 191], [0, 127, 255], [0, 100, 155]] +def stare_palette(): + """STARE palette for external use""" + return [[120, 120, 120], [6, 230, 230]] dataset_aliases = { 'cityscapes': ['cityscapes'], @@ -267,7 +273,8 @@ def isaid_palette(): 'coco-stuff10k', 'coco-stuff164k', 'coco_stuff', 'coco_stuff10k', 'coco_stuff164k' ], - 'isaid': ['isaid', 'iSAID'] + 'isaid': ['isaid', 'iSAID'], + 'stare':['stare', 'STARE'] } From c200bbff3b9646dca7cd35f9079c620c2dac09ff Mon Sep 17 00:00:00 2001 From: CCODING Date: Tue, 15 Mar 2022 21:04:21 +0800 Subject: [PATCH 2/2] Update mmseg/core/evaluation/class_names.py Co-authored-by: MengzhangLI --- mmseg/core/evaluation/class_names.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/core/evaluation/class_names.py b/mmseg/core/evaluation/class_names.py index f900e3ed3a..e450259631 100644 --- a/mmseg/core/evaluation/class_names.py +++ b/mmseg/core/evaluation/class_names.py @@ -258,7 +258,7 @@ def isaid_palette(): [0, 127, 255], [0, 100, 155]] def stare_palette(): - """STARE palette for external use""" + """STARE palette for external use.""" return [[120, 120, 120], [6, 230, 230]] dataset_aliases = {