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_aliasesmasterbak
parent
558ccf9e6a
commit
10979986a6
|
@ -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']
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue