Skip to content

Commit 35cd28b

Browse files
jinwonkim93MeowZheng
authored andcommitted
Support for occludedface for external use (open-mmlab#2463)
## Motivation Please describe the motivation of this PR and the goal you want to achieve through this PR. support get_classes, get_palette for Occluded Face dataset. ## Modification Please briefly describe what modification is made in this PR. add occludedface_classes() add occludedface_palette() modified dataset_aliases
1 parent a6ea630 commit 35cd28b

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

mmseg/core/evaluation/class_names.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,11 @@ def stare_classes():
126126
return ['background', 'vessel']
127127

128128

129+
def occludedface_classes():
130+
"""occludedface class names for external use."""
131+
return ['background', 'face']
132+
133+
129134
def cityscapes_palette():
130135
"""Cityscapes palette for external use."""
131136
return [[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156],
@@ -265,6 +270,11 @@ def stare_palette():
265270
return [[120, 120, 120], [6, 230, 230]]
266271

267272

273+
def occludedface_palette():
274+
"""occludedface palette for external use."""
275+
return [[0, 0, 0], [128, 0, 0]]
276+
277+
268278
dataset_aliases = {
269279
'cityscapes': ['cityscapes'],
270280
'ade': ['ade', 'ade20k'],
@@ -278,7 +288,8 @@ def stare_palette():
278288
'coco_stuff164k'
279289
],
280290
'isaid': ['isaid', 'iSAID'],
281-
'stare': ['stare', 'STARE']
291+
'stare': ['stare', 'STARE'],
292+
'occludedface': ['occludedface']
282293
}
283294

284295

0 commit comments

Comments
 (0)