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

[Enhancement] Support inherit cityscapes dataset (#750)

Browse Source 
* rewrite init function

* init function parameters
This commit is contained in:
谢昕辰 2021-08-27 14:53:36 +08:00 committed by GitHub
parent 7ccb9c3995
commit 4686f1ec28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
mmseg/datasets/cityscapes.py
View File

@ -29,11 +29,12 @@ class CityscapesDataset(CustomDataset):
[255, 0, 0], [0, 0, 142], [0, 0, 70], [0, 60, 100], [255, 0, 0], [0, 0, 142], [0, 0, 70], [0, 60, 100],
[0, 80, 100], [0, 0, 230], [119, 11, 32]] [0, 80, 100], [0, 0, 230], [119, 11, 32]]
def __init__(self, **kwargs): def __init__(self,
img_suffix='_leftImg8bit.png',
seg_map_suffix='_gtFine_labelTrainIds.png',
**kwargs):
super(CityscapesDataset, self).__init__( super(CityscapesDataset, self).__init__(
img_suffix='_leftImg8bit.png', img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs)
seg_map_suffix='_gtFine_labelTrainIds.png',
**kwargs)
@staticmethod @staticmethod
def _convert_to_label_id(result): def _convert_to_label_id(result):