@@ -4,22 +4,22 @@ defaults:
 
 train_dataset_params:
   transforms:
-    - ColorJitterSeg:
+    - SegColorJitter:
         brightness: 0.5
         contrast: 0.5
         saturation: 0.5
 
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.5, 2. ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: [ 1024, 1024 ]
         fill_mask: 19
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: [ 1024, 1024 ]
         mode: random
 

@@ -5,23 +5,23 @@ defaults:
 train_dataset_params:
   transforms:
     # for more options see common.factories.transforms_factory.py
-    - ColorJitterSeg:
+    - SegColorJitter:
         brightness: 0.1
         contrast: 0.1
         saturation: 0.1
 
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.4, 1.6 ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: 1024
         fill_image: [ 19, 0, 0 ]
         fill_mask: 19                     # ignored label idx
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 1024
         mode: random
 

@@ -5,28 +5,28 @@ defaults:
 train_dataset_params:
   transforms:
     # for more options see common.factories.transforms_factory.py
-    - ColorJitterSeg:
+    - SegColorJitter:
         brightness: 0.5
         contrast: 0.5
         saturation: 0.5
 
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.125, 1.5 ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: [ 1024, 512 ]
         fill_mask: 19                  # ignored label idx
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: [ 1024, 512 ]
         mode: random
 
 val_dataset_params:
   transforms:
-    - RescaleSeg:
+    - SegRescale:
         scale_factor: 0.5
 
 train_dataloader_params:

@@ -5,28 +5,28 @@ defaults:
 train_dataset_params:
   transforms:
     # for more options see common.factories.transforms_factory.py
-    - ColorJitterSeg:
+    - SegColorJitter:
         brightness: 0.5
         contrast: 0.5
         saturation: 0.5
 
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.125, 1.5 ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: [ 1536, 768 ]
         fill_mask: 19                  # ignored label idx
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: [ 1536, 768 ]
         mode: random
 
 val_dataset_params:
   transforms:
-    - RescaleSeg:
+    - SegRescale:
         scale_factor: 0.75
 
 train_dataloader_params:

@@ -9,19 +9,19 @@ train_dataset_params:
   cache_images: False
   transforms:
     # for more options see common.factories.transforms_factory.py
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RescaleSeg: # consider removing this step
+    - SegRescale: # consider removing this step
         long_size: 608
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.5, 2.0 ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: 512
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 512
         mode: random
 
@@ -35,10 +35,10 @@ val_dataset_params:
   cache_labels: False
   cache_images: False
   transforms:
-    - RescaleSeg:
+    - SegRescale:
         short_size: 512
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 512
         mode: center
 

@@ -7,19 +7,19 @@ train_dataset_params:
   cache_images: False
   transforms:
     # for more options see common.factories.transforms_factory.py
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RescaleSeg: # consider removing this step
+    - SegRescale: # consider removing this step
         long_size: 608
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.5, 2.0 ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: 512
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 512
         mode: random
 
@@ -31,10 +31,10 @@ val_dataset_params:
   cache_labels: False
   cache_images: False
   transforms:
-    - RescaleSeg:
+    - SegRescale:
         short_size: 512
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 512
         mode: center
 

@@ -10,19 +10,19 @@ train_dataset_params:
   cache_images: False
   transforms:
     # for more options see common.factories.transforms_factory.py
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
 
-    - RescaleSeg: # consider removing this step
+    - SegRescale: # consider removing this step
         long_size: 608
 
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.5, 2.0 ]
 
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: 512
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 512
         mode: random
 
@@ -34,10 +34,10 @@ val_dataset_params:
   cache_labels: False
   cache_images: False
   transforms:
-    - RescaleSeg:
+    - SegRescale:
         short_size: 512
 
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: 512
         mode: center
 

@@ -4,18 +4,18 @@ train_dataset_params:
   cache_labels: False
   cache_images: False
   transforms:
-    - RandomRescaleSeg:
+    - SegRandomRescale:
         scales: [ 0.25, 1. ]
-    - ColorJitterSeg:
+    - SegColorJitter:
         brightness: 0.5
         contrast: 0.5
         saturation: 0.5
-    - RandomFlipSeg:
+    - SegRandomFlip:
         prob: 0.5
-    - PadShortToCropSizeSeg:
+    - SegPadShortToCropSize:
         crop_size: [ 320, 480 ]
         fill_mask: 0
-    - CropImageAndMaskSeg:
+    - SegCropImageAndMask:
         crop_size: [ 320, 480 ]
         mode: random
 
@@ -25,7 +25,7 @@ val_dataset_params:
   cache_labels: False
   cache_images: False
   transforms:
-    - ResizeSeg:
+    - SegResize:
         h: 480
         w: 320