[Fix]Add input argument for datasets (#2379)

mmseg/datasets/ade.py

@@ -80,9 +80,13 @@ class ADE20KDataset(BaseSegDataset):
  [184, 255, 0], [0, 133, 255], [255, 214, 0], [25, 194, 194],
  [102, 255, 0], [92, 0, 255]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.jpg',
+ seg_map_suffix='.png',
+ reduce_zero_label=True,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.jpg',
- seg_map_suffix='.png',
- reduce_zero_label=True,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)

mmseg/datasets/chase_db1.py

@@ -17,10 +17,14 @@ class ChaseDB1Dataset(BaseSegDataset):
  classes=('background', 'vessel'),
  palette=[[120, 120, 120], [6, 230, 230]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='_1stHO.png',
+ reduce_zero_label=False,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='_1stHO.png',
- reduce_zero_label=False,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)
  assert self.file_client.exists(self.data_prefix['img_path'])

mmseg/datasets/coco_stuff.py

@@ -91,6 +91,9 @@ class COCOStuffDataset(BaseSegDataset):
  [192, 192, 0], [128, 64, 96], [192, 32, 64], [192, 64, 128],
  [64, 192, 96], [64, 160, 64], [64, 64, 0]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.jpg',
+ seg_map_suffix='_labelTrainIds.png',
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.jpg', seg_map_suffix='_labelTrainIds.png', **kwargs)
+ img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs)

mmseg/datasets/dark_zurich.py

@@ -7,8 +7,9 @@
 class DarkZurichDataset(CityscapesDataset):
  """DarkZurichDataset dataset."""
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='_rgb_anon.png',
+ seg_map_suffix='_gt_labelTrainIds.png',
+ **kwargs) -> None:
  super().__init__(
- img_suffix='_rgb_anon.png',
- seg_map_suffix='_gt_labelTrainIds.png',
- **kwargs)
+ img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs)

mmseg/datasets/drive.py

@@ -17,10 +17,14 @@ class DRIVEDataset(BaseSegDataset):
  classes=('background', 'vessel'),
  palette=[[120, 120, 120], [6, 230, 230]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='_manual1.png',
+ reduce_zero_label=False,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='_manual1.png',
- reduce_zero_label=False,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)
  assert self.file_client.exists(self.data_prefix['img_path'])

mmseg/datasets/hrf.py

@@ -17,10 +17,14 @@ class HRFDataset(BaseSegDataset):
  classes=('background', 'vessel'),
  palette=[[120, 120, 120], [6, 230, 230]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='.png',
+ reduce_zero_label=False,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='.png',
- reduce_zero_label=False,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)
  assert self.file_client.exists(self.data_prefix['img_path'])

mmseg/datasets/isaid.py

@@ -23,10 +23,14 @@ class iSAIDDataset(BaseSegDataset):
  [0, 0, 127], [0, 0, 191], [0, 0, 255], [0, 191, 127],
  [0, 127, 191], [0, 127, 255], [0, 100, 155]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='_instance_color_RGB.png',
+ ignore_index=255,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='_instance_color_RGB.png',
- ignore_index=255,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ ignore_index=ignore_index,
  **kwargs)
  assert self.file_client.exists(self.data_prefix['img_path'])

mmseg/datasets/isprs.py

@@ -17,9 +17,13 @@ class ISPRSDataset(BaseSegDataset):
  palette=[[255, 255, 255], [0, 0, 255], [0, 255, 255], [0, 255, 0],
  [255, 255, 0], [255, 0, 0]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='.png',
+ reduce_zero_label=True,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='.png',
- reduce_zero_label=True,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)

mmseg/datasets/lip.py

@@ -39,5 +39,9 @@ class LIPDataset(BaseSegDataset):
  [255, 170, 0],
  ))
 
- def __init__(self, **kwargs) -> None:
- super().__init__(img_suffix='.jpg', seg_map_suffix='.png', **kwargs)
+ def __init__(self,
+ img_suffix='.jpg',
+ seg_map_suffix='.png',
+ **kwargs) -> None:
+ super().__init__(
+ img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs)

mmseg/datasets/loveda.py

@@ -17,9 +17,13 @@ class LoveDADataset(BaseSegDataset):
  palette=[[255, 255, 255], [255, 0, 0], [255, 255, 0], [0, 0, 255],
  [159, 129, 183], [0, 255, 0], [255, 195, 128]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='.png',
+ reduce_zero_label=True,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='.png',
- reduce_zero_label=True,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)

mmseg/datasets/night_driving.py

@@ -7,8 +7,9 @@
 class NightDrivingDataset(CityscapesDataset):
  """NightDrivingDataset dataset."""
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='_leftImg8bit.png',
+ seg_map_suffix='_gtCoarse_labelTrainIds.png',
+ **kwargs) -> None:
  super().__init__(
- img_suffix='_leftImg8bit.png',
- seg_map_suffix='_gtCoarse_labelTrainIds.png',
- **kwargs)
+ img_suffix=img_suffix, seg_map_suffix=seg_map_suffix, **kwargs)

mmseg/datasets/pascal_context.py

@@ -45,10 +45,14 @@ class PascalContextDataset(BaseSegDataset):
  [255, 31, 0], [255, 224, 0], [153, 255, 0], [0, 0, 255],
  [255, 71, 0], [0, 235, 255], [0, 173, 255], [31, 0, 255]])
 
- def __init__(self, ann_file: str, **kwargs) -> None:
+ def __init__(self,
+ ann_file: str,
+ img_suffix='.jpg',
+ seg_map_suffix='.png',
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.jpg',
- seg_map_suffix='.png',
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
  ann_file=ann_file,
  reduce_zero_label=False,
  **kwargs)
@@ -95,12 +99,17 @@ class PascalContextDataset59(BaseSegDataset):
  [255, 31, 0], [255, 224, 0], [153, 255, 0], [0, 0, 255],
  [255, 71, 0], [0, 235, 255], [0, 173, 255], [31, 0, 255]])
 
- def __init__(self, ann_file: str, **kwargs):
+ def __init__(self,
+ ann_file: str,
+ img_suffix='.jpg',
+ seg_map_suffix='.png',
+ reduce_zero_label=True,
+ **kwargs):
  super().__init__(
- img_suffix='.jpg',
- seg_map_suffix='.png',
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
  ann_file=ann_file,
- reduce_zero_label=True,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)
  assert self.file_client.exists(
  self.data_prefix['img_path']) and osp.isfile(self.ann_file)

mmseg/datasets/potsdam.py

@@ -17,9 +17,13 @@ class PotsdamDataset(BaseSegDataset):
  palette=[[255, 255, 255], [0, 0, 255], [0, 255, 255], [0, 255, 0],
  [255, 255, 0], [255, 0, 0]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='.png',
+ reduce_zero_label=True,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='.png',
- reduce_zero_label=True,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)

mmseg/datasets/stare.py

@@ -16,10 +16,14 @@ class STAREDataset(BaseSegDataset):
  classes=('background', 'vessel'),
  palette=[[120, 120, 120], [6, 230, 230]])
 
- def __init__(self, **kwargs) -> None:
+ def __init__(self,
+ img_suffix='.png',
+ seg_map_suffix='.ah.png',
+ reduce_zero_label=False,
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.png',
- seg_map_suffix='.ah.png',
- reduce_zero_label=False,
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
+ reduce_zero_label=reduce_zero_label,
  **kwargs)
  assert self.file_client.exists(self.data_prefix['img_path'])

mmseg/datasets/voc.py

@@ -24,10 +24,14 @@ class PascalVOCDataset(BaseSegDataset):
  [0, 64, 0], [128, 64, 0], [0, 192, 0], [128, 192, 0],
  [0, 64, 128]])
 
- def __init__(self, ann_file, **kwargs) -> None:
+ def __init__(self,
+ ann_file,
+ img_suffix='.jpg',
+ seg_map_suffix='.png',
+ **kwargs) -> None:
  super().__init__(
- img_suffix='.jpg',
- seg_map_suffix='.png',
+ img_suffix=img_suffix,
+ seg_map_suffix=seg_map_suffix,
  ann_file=ann_file,
  **kwargs)
  assert self.file_client.exists(