Skip to content

Commit 3d0c2eb

Browse files
Support open-mmlab#1375: add demo/image_demo.py support for STARE (open-mmlab#1376)
* Support open-mmlab#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]>
1 parent a3aab38 commit 3d0c2eb

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

mmseg/core/evaluation/class_names.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,9 @@ def isaid_classes():
120120
'Soccer_ball_field', 'plane', 'Harbor'
121121
]
122122

123+
def stare_classes():
124+
"""stare class names for external use."""
125+
return ['background', 'vessel']
123126

124127
def cityscapes_palette():
125128
"""Cityscapes palette for external use."""
@@ -254,6 +257,9 @@ def isaid_palette():
254257
[0, 0, 191], [0, 0, 255], [0, 191, 127], [0, 127, 191],
255258
[0, 127, 255], [0, 100, 155]]
256259

260+
def stare_palette():
261+
"""STARE palette for external use."""
262+
return [[120, 120, 120], [6, 230, 230]]
257263

258264
dataset_aliases = {
259265
'cityscapes': ['cityscapes'],
@@ -267,7 +273,8 @@ def isaid_palette():
267273
'coco-stuff10k', 'coco-stuff164k', 'coco_stuff', 'coco_stuff10k',
268274
'coco_stuff164k'
269275
],
270-
'isaid': ['isaid', 'iSAID']
276+
'isaid': ['isaid', 'iSAID'],
277+
'stare':['stare', 'STARE']
271278
}
272279

273280

0 commit comments

Comments
 (0)