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

[Fix] Fix ignore class id from -1 to 255 in BaseSegDataset in 1.x

Browse Source
This commit is contained in:
MengzhangLI 2022-11-22 00:52:36 +08:00
parent 8ea777e245
commit cfab5b39c0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmseg/datasets/basesegdataset.py
View File

@ -179,7 +179,7 @@ class BaseSegDataset(BaseDataset):
f'subset of classes {old_classes} in METAINFO.') f'subset of classes {old_classes} in METAINFO.')
for i, c in enumerate(old_classes): for i, c in enumerate(old_classes):
if c not in new_classes: if c not in new_classes:
label_map[i] = -1 label_map[i] = 255
else: else:
label_map[i] = new_classes.index(c) label_map[i] = new_classes.index(c)
return label_map return label_map