mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

Support #1375: add demo/image_demo.py support for STARE (#1376)

Browse Source 
* Support #1375: add demo/image_demo.py support for STARE

* Update mmseg/core/evaluation/class_names.py

Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>

Co-authored-by: MengzhangLI <mcmong@pku.edu.cn>
This commit is contained in:
CCODING 2022-03-15 22:42:20 +08:00 committed by GitHub
parent a3aab38bb4
commit 3d0c2eb385
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmseg/core/evaluation/class_names.py
View File

@ -120,6 +120,9 @@ def isaid_classes():
'Soccer_ball_field', 'plane', 'Harbor' 'Soccer_ball_field', 'plane', 'Harbor'
] ]
def stare_classes():
"""stare class names for external use."""
return ['background', 'vessel']
def cityscapes_palette(): def cityscapes_palette():
"""Cityscapes palette for external use.""" """Cityscapes palette for external use."""
@ -254,6 +257,9 @@ def isaid_palette():
[0, 0, 191], [0, 0, 255], [0, 191, 127], [0, 127, 191], [0, 0, 191], [0, 0, 255], [0, 191, 127], [0, 127, 191],
[0, 127, 255], [0, 100, 155]] [0, 127, 255], [0, 100, 155]]
def stare_palette():
"""STARE palette for external use."""
return [[120, 120, 120], [6, 230, 230]]
dataset_aliases = { dataset_aliases = {
'cityscapes': ['cityscapes'], 'cityscapes': ['cityscapes'],
@ -267,7 +273,8 @@ dataset_aliases = {
'coco-stuff10k', 'coco-stuff164k', 'coco_stuff', 'coco_stuff10k', 'coco-stuff10k', 'coco-stuff164k', 'coco_stuff', 'coco_stuff10k',
'coco_stuff164k' 'coco_stuff164k'
], ],
'isaid': ['isaid', 'iSAID'] 'isaid': ['isaid', 'iSAID'],
'stare':['stare', 'STARE']
} }