Support for occludedface for external use (#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
masterbak
whooray 2023-01-11 12:52:46 +09:00 committed by GitHub
parent 558ccf9e6a
commit 10979986a6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 1 deletions

View File

@ -126,6 +126,11 @@ def stare_classes():
return ['background', 'vessel'] return ['background', 'vessel']
def occludedface_classes():
"""occludedface class names for external use."""
return ['background', 'face']
def cityscapes_palette(): def cityscapes_palette():
"""Cityscapes palette for external use.""" """Cityscapes palette for external use."""
return [[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156], return [[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156],
@ -265,6 +270,11 @@ def stare_palette():
return [[120, 120, 120], [6, 230, 230]] return [[120, 120, 120], [6, 230, 230]]
def occludedface_palette():
"""occludedface palette for external use."""
return [[0, 0, 0], [128, 0, 0]]
dataset_aliases = { dataset_aliases = {
'cityscapes': ['cityscapes'], 'cityscapes': ['cityscapes'],
'ade': ['ade', 'ade20k'], 'ade': ['ade', 'ade20k'],
@ -278,7 +288,8 @@ dataset_aliases = {
'coco_stuff164k' 'coco_stuff164k'
], ],
'isaid': ['isaid', 'iSAID'], 'isaid': ['isaid', 'iSAID'],
'stare': ['stare', 'STARE'] 'stare': ['stare', 'STARE'],
'occludedface': ['occludedface']
} }