Support #1375: add demo/image_demo.py support for STARE (#1376)

* Support #1375: add demo/image_demo.py support for STARE * Update mmseg/core/evaluation/class_names.py Co-authored-by: MengzhangLI <[email protected]> Co-authored-by: MengzhangLI <[email protected]>
open-mmlab · Mar 15, 2022 · 3d0c2eb · 3d0c2eb
1 parent a3aab38
commit 3d0c2eb
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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']
 }