[configs.datasets] DRY w.r.t. data augmentation

bob/ip/binseg/configs/datasets/chasedb1.py

@@ -23,14 +23,9 @@ bright strip running down the centre known as the central vessel reflex.
 * Split reference: [CHASEDB1-2012]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Crop(0, 18, 960, 960),  #(upper, left, height, width)
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Crop
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Crop(0, 18, 960, 960)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.chasedb1 import dataset as chasedb1

bob/ip/binseg/configs/datasets/chasedb1544.py

-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Resize(544),
-        Crop(0, 12, 544, 544),
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Resize, Crop
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Resize(544), Crop(0, 12, 544, 544)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.chasedb1 import dataset as chasedb1

bob/ip/binseg/configs/datasets/csv.py

@@ -67,14 +67,11 @@ See:
 """
 
 # add your transforms below - these are just examples
-from bob.ip.binseg.data.transforms import *
+from bob.ip.binseg.data.transforms import CenterCrop
+#from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
 _transforms = [
-        #CenterCrop((544, 544)),
-        #RandomHFlip(),
-        #RandomVFlip(),
-        #RandomRotation(),
-        #ColorJitter(),
-        ]
+        CenterCrop((544, 544)),
+        ] # + _DA
 
 from bob.ip.binseg.data.csvdataset import CSVDataset
 #dataset = CSVDataset("<path.csv>", check_available=False, transforms=_transforms)

bob/ip/binseg/configs/datasets/drionsdb.py

@@ -18,23 +18,12 @@ baseline.
 * Split reference: [MANINIS-2016]_
 """
 
+from bob.ip.binseg.data.transforms import Pad
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Pad((4, 8, 4, 8))] + _DA
 
 from bob.db.drionsdb import Database as DRIONS
-from bob.ip.binseg.data.transforms import *
-from bob.ip.binseg.data.binsegdataset import BinSegDataset
-
-#### Config ####
-
-_transforms = [
-        Pad((4, 8, 4, 8)),  #(left, top, right, bottom)
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
-
-# bob.db.dataset init
 bobdb = DRIONS(protocol="default")
 
-# PyTorch dataset
+from bob.ip.binseg.data.binsegdataset import BinSegDataset
 dataset = BinSegDataset(bobdb, split="train", transforms=_transforms)

bob/ip/binseg/configs/datasets/drionsdb_test.py

@@ -18,16 +18,11 @@ baseline.
 * Split reference: [MANINIS-2016]_
 """
 
-from bob.db.drionsdb import Database as DRIONS
-from bob.ip.binseg.data.transforms import *
-from bob.ip.binseg.data.binsegdataset import BinSegDataset
-
-#### Config ####
-
+from bob.ip.binseg.data.transforms import Pad
 _transforms = [Pad((4, 8, 4, 8))]
 
-# bob.db.dataset init
+from bob.db.drionsdb import Database as DRIONS
 bobdb = DRIONS(protocol="default")
 
-# PyTorch dataset
+from bob.ip.binseg.data.binsegdataset import BinSegDataset
 dataset = BinSegDataset(bobdb, split="test", transforms=_transforms)

bob/ip/binseg/configs/datasets/dristhigs1_cup.py

@@ -21,22 +21,12 @@ and notching information.
 * Split reference: [DRISHTIGS1-2014]_
 """
 
-from bob.db.drishtigs1 import Database as DRISHTI
-from bob.ip.binseg.data.transforms import *
-from bob.ip.binseg.data.binsegdataset import BinSegDataset
-
-#### Config ####
+from bob.ip.binseg.data.transforms import CenterCrop
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [CenterCrop((1760, 2048))] + _DA
 
-_transforms = [
-        CenterCrop((1760, 2048)),
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
-
-# bob.db.dataset init
+from bob.db.drishtigs1 import Database as DRISHTI
 bobdb = DRISHTI(protocol="default_cup")
 
-# PyTorch dataset
+from bob.ip.binseg.data.binsegdataset import BinSegDataset
 dataset = BinSegDataset(bobdb, split="train", transforms=_transforms)

bob/ip/binseg/configs/datasets/dristhigs1_cup_test.py

@@ -21,16 +21,11 @@ and notching information.
 * Split reference: [DRISHTIGS1-2014]_
 """
 
-from bob.db.drishtigs1 import Database as DRISHTI
 from bob.ip.binseg.data.transforms import CenterCrop
-from bob.ip.binseg.data.binsegdataset import BinSegDataset
-
-#### Config ####
-
 _transforms = [CenterCrop((1760, 2048))]
 
-# bob.db.dataset init
+from bob.db.drishtigs1 import Database as DRISHTI
 bobdb = DRISHTI(protocol="default_cup")
 
-# PyTorch dataset
+from bob.ip.binseg.data.binsegdataset import BinSegDataset
 dataset = BinSegDataset(bobdb, split="test", transforms=_transforms)

bob/ip/binseg/configs/datasets/dristhigs1_od.py

@@ -21,22 +21,12 @@ and notching information.
 * Split reference: [DRISHTIGS1-2014]_
 """
 
-from bob.db.drishtigs1 import Database as DRISHTI
-from bob.ip.binseg.data.transforms import *
-from bob.ip.binseg.data.binsegdataset import BinSegDataset
-
-#### Config ####
+from bob.ip.binseg.data.transforms import CenterCrop
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [CenterCrop((1760, 2048))] + _DA
 
-_transforms = [
-        CenterCrop((1760, 2048)),
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
-
-# bob.db.dataset init
+from bob.db.drishtigs1 import Database as DRISHTI
 bobdb = DRISHTI(protocol="default_od")
 
-# PyTorch dataset
+from bob.ip.binseg.data.binsegdataset import BinSegDataset
 dataset = BinSegDataset(bobdb, split="train", transforms=_transforms)

bob/ip/binseg/configs/datasets/dristhigs1_od_test.py

@@ -21,16 +21,11 @@ and notching information.
 * Split reference: [DRISHTIGS1-2014]_
 """
 
-from bob.db.drishtigs1 import Database as DRISHTI
 from bob.ip.binseg.data.transforms import CenterCrop
-from bob.ip.binseg.data.binsegdataset import BinSegDataset
-
-#### Config ####
-
 _transforms = [CenterCrop((1760, 2048))]
 
-# bob.db.dataset init
+from bob.db.drishtigs1 import Database as DRISHTI
 bobdb = DRISHTI(protocol="default_od")
 
-# PyTorch dataset
+from bob.ip.binseg.data.binsegdataset import BinSegDataset
 dataset = BinSegDataset(bobdb, split="test", transforms=_transforms)

bob/ip/binseg/configs/datasets/drive.py

@@ -13,14 +13,9 @@ segmentation of blood vessels in retinal images.
 * Split reference: [DRIVE-2004]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        CenterCrop((544, 544)),
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import CenterCrop
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [CenterCrop((544, 544))] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.drive import dataset as drive

bob/ip/binseg/configs/datasets/folder.py

@@ -37,7 +37,7 @@ possible.
 """
 
 # add your transforms below - these are just examples
-from bob.ip.binseg.data.transforms import *
+from bob.ip.binseg.data.transforms import CenterCrop
 _transforms = [
         #CenterCrop((544, 544)),
     ]

bob/ip/binseg/configs/datasets/hrf.py

@@ -14,14 +14,9 @@ x 2336. One set of ground-truth vessel annotations is available.
 * Split reference: [ORLANDO-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Crop(0, 108, 2336, 3296),
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Crop
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Crop(0, 108, 2336, 3296)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.hrf import dataset as hrf

bob/ip/binseg/configs/datasets/hrf1024.py

@@ -14,15 +14,9 @@ x 2336. One set of ground-truth vessel annotations is available.
 * Split reference: [ORLANDO-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Pad((0, 584, 0, 584)),
-        Resize(1024),
-        RandomRotation(),
-        RandomHFlip(),
-        RandomVFlip(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Pad, Resize
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Pad((0, 584, 0, 584)), Resize(1024)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.hrf import dataset as hrf

bob/ip/binseg/configs/datasets/hrf544.py

@@ -14,15 +14,9 @@ x 2336. One set of ground-truth vessel annotations is available.
 * Split reference: [ORLANDO-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Resize((363)),
-        Pad((0, 90, 0, 91)),
-        RandomRotation(),
-        RandomHFlip(),
-        RandomVFlip(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Resize, Pad
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Resize((363)), Pad((0, 90, 0, 91))] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.hrf import dataset as hrf

bob/ip/binseg/configs/datasets/hrf608.py

@@ -14,15 +14,9 @@ x 2336. One set of ground-truth vessel annotations is available.
 * Split reference: [ORLANDO-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Pad((0, 345, 0, 345)),
-        Resize(608),
-        RandomRotation(),
-        RandomHFlip(),
-        RandomVFlip(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Pad, Resize
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Pad((0, 345, 0, 345)), Resize(608)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.hrf import dataset as hrf

bob/ip/binseg/configs/datasets/hrf960.py

@@ -14,15 +14,9 @@ x 2336. One set of ground-truth vessel annotations is available.
 * Split reference: [ORLANDO-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Pad((0, 584, 0, 584)),
-        Resize(960),
-        RandomRotation(),
-        RandomHFlip(),
-        RandomVFlip(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Pad, Resize
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Pad((0, 584, 0, 584)), Resize(960)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.hrf import dataset as hrf

bob/ip/binseg/configs/datasets/hrf_1168.py

@@ -14,15 +14,9 @@ x 2336. One set of ground-truth vessel annotations is available.
 * Split reference: [ORLANDO-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Crop(0, 108, 2336, 3296),  #(upper, left, height, width)
-        Resize(1168),  # applies to the smaller edge
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Crop, Resize
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Crop(0, 108, 2336, 3296), Resize(1168)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.hrf import dataset as hrf

bob/ip/binseg/configs/datasets/iostar_od.py

@@ -16,14 +16,8 @@ dataset includes annotations for the optic disc and the artery/vein ratio.
 """
 
 from bob.ip.binseg.data.transforms import *
-_transforms = [
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
-
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.iostar import dataset as iostar
 dataset = SampleList2TorchDataset(iostar.subsets("optic-disc")["train"],
-        transforms=_transforms)
+        transforms=_DA)

bob/ip/binseg/configs/datasets/iostar_vessel.py

@@ -16,14 +16,8 @@ dataset includes annotations for the optic disc and the artery/vein ratio.
 """
 
 from bob.ip.binseg.data.transforms import *
-_transforms = [
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
-
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.iostar import dataset as iostar
 dataset = SampleList2TorchDataset(iostar.subsets("vessel")["train"],
-        transforms=_transforms)
+        transforms=_DA)

bob/ip/binseg/configs/datasets/iostarvessel544.py

@@ -15,14 +15,9 @@ dataset includes annotations for the optic disc and the artery/vein ratio.
 * Split reference: [MEYER-2017]_
 """
 
-from bob.ip.binseg.data.transforms import *
-_transforms = [
-        Resize(544),
-        RandomHFlip(),
-        RandomVFlip(),
-        RandomRotation(),
-        ColorJitter(),
-        ]
+from bob.ip.binseg.data.transforms import Resize
+from bob.ip.binseg.configs.datasets.utils import DATA_AUGMENTATION as _DA
+_transforms = [Resize(544)] + _DA
 
 from bob.ip.binseg.data.utils import SampleList2TorchDataset
 from bob.ip.binseg.data.iostar import dataset as iostar