Skip to content

Commit

Permalink
Support for occludedface for external use (#2463)
Browse files Browse the repository at this point in the history
## 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
  • Loading branch information
jinwonkim93 committed Jan 11, 2023
1 parent 558ccf9 commit 1097998
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion mmseg/core/evaluation/class_names.py
Expand Up @@ -126,6 +126,11 @@ def stare_classes():
return ['background', 'vessel']


def occludedface_classes():
"""occludedface class names for external use."""
return ['background', 'face']


def cityscapes_palette():
"""Cityscapes palette for external use."""
return [[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156],
Expand Down Expand Up @@ -265,6 +270,11 @@ def stare_palette():
return [[120, 120, 120], [6, 230, 230]]


def occludedface_palette():
"""occludedface palette for external use."""
return [[0, 0, 0], [128, 0, 0]]


dataset_aliases = {
'cityscapes': ['cityscapes'],
'ade': ['ade', 'ade20k'],
Expand All @@ -278,7 +288,8 @@ def stare_palette():
'coco_stuff164k'
],
'isaid': ['isaid', 'iSAID'],
'stare': ['stare', 'STARE']
'stare': ['stare', 'STARE'],
'occludedface': ['occludedface']
}


Expand Down

0 comments on commit 1097998

Please sign in to comment.