Skip to content
Snippets Groups Projects
Commit 42c3f944 authored by 谢昕辰's avatar 谢昕辰 Committed by GitHub
Browse files

[Enhancement] Support inherit cityscapes dataset (#750)

* rewrite init function

* init function parameters
parent 45707744
No related branches found
No related tags found
No related merge requests found
...@@ -29,11 +29,12 @@ class CityscapesDataset(CustomDataset): ...@@ -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):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment