diff --git a/bob/ip/binseg/configs/datasets/chasedb11024all.py b/bob/ip/binseg/configs/datasets/chasedb11024all.py new file mode 100644 index 0000000000000000000000000000000000000000..39d585cb513fcf380911180c1a4fe171b0139c6f --- /dev/null +++ b/bob/ip/binseg/configs/datasets/chasedb11024all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.chasedb1 import Database as CHASEDB1 +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Crop(0,18,960,960) + ,Resize(1024) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = CHASEDB1(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/chasedb11168all.py b/bob/ip/binseg/configs/datasets/chasedb11168all.py new file mode 100644 index 0000000000000000000000000000000000000000..e30c1ef7016310303435cffebe0cfdad67a066f7 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/chasedb11168all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.chasedb1 import Database as CHASEDB1 +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Crop(140,18,680,960) + ,Resize(1168) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = CHASEDB1(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive1024all.py b/bob/ip/binseg/configs/datasets/drive1024all.py new file mode 100644 index 0000000000000000000000000000000000000000..26d82bc8652ec5d8c474e859bea67d100d571112 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive1024all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.drive import Database as DRIVE +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + RandomRotation() + ,CenterCrop((540,540)) + ,Resize(1024) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive1168all.py b/bob/ip/binseg/configs/datasets/drive1168all.py new file mode 100644 index 0000000000000000000000000000000000000000..f1e9d8562460970bd6c55222db7658c5d6fd02a4 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive1168all.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.drive import Database as DRIVE +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Crop(75,10,416,544) + ,Pad((21,0,22,0)) + ,Resize(1168) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive960all.py b/bob/ip/binseg/configs/datasets/drive960all.py new file mode 100644 index 0000000000000000000000000000000000000000..1242574ba3eef4935cbc6b3f22e9ddf8ed3a1ef0 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive960all.py @@ -0,0 +1,27 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.drive import Database as DRIVE +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch +#### Config #### + +transforms = Compose([ + RandomRotation() + ,CenterCrop((544,544)) + ,Resize(960) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024ssliostar.py b/bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024ssliostar.py index 4dbd23d2cf52f40e4b03c38bb603f4139900a1c1..dd9016b7c076b08cc99e875d4f5cc9dcdd3babd0 100644 --- a/bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024ssliostar.py +++ b/bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024ssliostar.py @@ -13,10 +13,9 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, Un # PyTorch dataset labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,chasedb]) -#### Unlabeled CHASE TRAIN #### +#### Unlabeled IOSTAR Train #### unlabeled_transforms = Compose([ - Crop(0,18,960,960) - ,RandomHFlip() + RandomHFlip() ,RandomVFlip() ,RandomRotation() ,ColorJitter() diff --git a/bob/ip/binseg/configs/datasets/drivestarechasedb1hrfl1024_all_ssliostar.py b/bob/ip/binseg/configs/datasets/drivestarechasedb1hrfl1024_all_ssliostar.py new file mode 100644 index 0000000000000000000000000000000000000000..7e68cb52cc45a477cfb33abe8b8efbd019e4c8cd --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drivestarechasedb1hrfl1024_all_ssliostar.py @@ -0,0 +1,33 @@ +from bob.ip.binseg.configs.datasets.drive1024all import dataset as drive +from bob.ip.binseg.configs.datasets.stare1024all import dataset as stare +from bob.ip.binseg.configs.datasets.hrf1024all import dataset as hrf +from bob.ip.binseg.configs.datasets.chasedb11024all import dataset as chasedb +from bob.db.iostar import Database as IOSTAR +from bob.ip.binseg.data.transforms import * +import torch +from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset + + +#### Config #### + +# PyTorch dataset +labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,chasedb]) + +#### Unlabeled IOSTAR Train #### +unlabeled_transforms = Compose([ + RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +iostarbobdb = IOSTAR(protocol='default_vessel') + +# PyTorch dataset +unlabeled_dataset = UnLabeledBinSegDataset(iostarbobdb, split='train', transform=unlabeled_transforms) + +# SSL Dataset + +dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168_all_sslhrf.py b/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168_all_sslhrf.py new file mode 100644 index 0000000000000000000000000000000000000000..fde74696dc7be96eb77c0870fa9475755b181e3b --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168_all_sslhrf.py @@ -0,0 +1,35 @@ +from bob.ip.binseg.configs.datasets.drive1168all import dataset as drive +from bob.ip.binseg.configs.datasets.stare1168all import dataset as stare +from bob.ip.binseg.configs.datasets.chasedb11168all import dataset as chasedb +from bob.ip.binseg.configs.datasets.iostarvessel1168all import dataset as iostar +from bob.db.hrf import Database as HRF +from bob.ip.binseg.data.transforms import * +import torch +from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset + + +#### Config #### + +# PyTorch dataset +labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,iostar,chasedb]) + +#### Unlabeled HRF TRAIN #### +unlabeled_transforms = Compose([ + RandomRotation() + ,Crop(0,108,2336,3296) + ,Resize((1168)) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +hrfbobdb = HRF(protocol='default') + +# PyTorch dataset +unlabeled_dataset = UnLabeledBinSegDataset(hrfbobdb, split='train', transform=unlabeled_transforms) + +# SSL Dataset + +dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168sslhrf.py b/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168sslhrf.py index 376aae1482ac3c8bd6911ae61e83f498bb5ad6c6..01705e15d8752b628ed6b150cf095db08ca4eed6 100644 --- a/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168sslhrf.py +++ b/bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168sslhrf.py @@ -13,13 +13,13 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, Un # PyTorch dataset labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,iostar,chasedb]) -#### Unlabeled CHASE TRAIN #### +#### Unlabeled HRF TRAIN #### unlabeled_transforms = Compose([ - Crop(0,108,2336,3296) + RandomRotation() + ,Crop(0,108,2336,3296) ,Resize((1168)) ,RandomHFlip() ,RandomVFlip() - ,RandomRotation() ,ColorJitter() ,ToTensor() ]) diff --git a/bob/ip/binseg/configs/datasets/drivestareiostarhrf960_all_sslchase.py b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960_all_sslchase.py new file mode 100644 index 0000000000000000000000000000000000000000..5653591b5b02c5e6af6c92156e9dffb9c00011a2 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960_all_sslchase.py @@ -0,0 +1,35 @@ +from bob.ip.binseg.configs.datasets.drive960all import dataset as drive +from bob.ip.binseg.configs.datasets.stare960all import dataset as stare +from bob.ip.binseg.configs.datasets.hrf960all import dataset as hrf +from bob.ip.binseg.configs.datasets.iostarvessel960all import dataset as iostar +from bob.db.chasedb1 import Database as CHASE +from bob.db.hrf import Database as HRF +from bob.ip.binseg.data.transforms import * +import torch +from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset + + +#### Config #### + +# PyTorch dataset +labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,iostar]) + +#### Unlabeled CHASE TRAIN #### +unlabeled_transforms = Compose([ + Crop(0,18,960,960) + ,RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +chasebobdb = CHASE(protocol = 'default') + +# PyTorch dataset +unlabeled_dataset = UnLabeledBinSegDataset(chasebobdb, split='train', transform=unlabeled_transforms) + +# SSL Dataset + +dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/hrf1024all.py b/bob/ip/binseg/configs/datasets/hrf1024all.py new file mode 100644 index 0000000000000000000000000000000000000000..c5fc4de232e87a8b77f979335058c64ff62ce7c5 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/hrf1024all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.hrf import Database as HRF +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + Pad((0,584,0,584)) + ,Resize((1024)) + ,RandomRotation() + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = HRF(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) diff --git a/bob/ip/binseg/configs/datasets/hrf960all.py b/bob/ip/binseg/configs/datasets/hrf960all.py new file mode 100644 index 0000000000000000000000000000000000000000..a34b3090520e94393c7fcbb8185bd62246954733 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/hrf960all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.hrf import Database as HRF +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + Pad((0,584,0,584)) + ,Resize((960)) + ,RandomRotation() + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = HRF(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) diff --git a/bob/ip/binseg/configs/datasets/iostarvessel1168all.py b/bob/ip/binseg/configs/datasets/iostarvessel1168all.py new file mode 100644 index 0000000000000000000000000000000000000000..b97d1ed0491458eef7b9c3c36dd472c1d4445b2c --- /dev/null +++ b/bob/ip/binseg/configs/datasets/iostarvessel1168all.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.iostar import Database as IOSTAR +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Crop(144,0,768,1024) + ,Pad((30,0,30,0)) + ,Resize(1168) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = IOSTAR(protocol='default_vessel') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/iostarvessel960all.py b/bob/ip/binseg/configs/datasets/iostarvessel960all.py new file mode 100644 index 0000000000000000000000000000000000000000..1338dc225b9cb1fc4f702fafdf8ed991b9dec5fd --- /dev/null +++ b/bob/ip/binseg/configs/datasets/iostarvessel960all.py @@ -0,0 +1,27 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.iostar import Database as IOSTAR +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + Resize(960) + ,RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = IOSTAR(protocol='default_vessel') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/stare1024all.py b/bob/ip/binseg/configs/datasets/stare1024all.py new file mode 100644 index 0000000000000000000000000000000000000000..46d8084985c90debdc39ae392c9e358aa11f19fd --- /dev/null +++ b/bob/ip/binseg/configs/datasets/stare1024all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.stare import Database as STARE +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Pad((0,32,0,32)) + ,Resize(1024) + ,CenterCrop(1024) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = STARE(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/stare1168all.py b/bob/ip/binseg/configs/datasets/stare1168all.py new file mode 100644 index 0000000000000000000000000000000000000000..d2384a0970c9384d494c20e969a1bc0ec916b1b9 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/stare1168all.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.stare import Database as STARE +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Crop(50,0,500,705) + ,Resize(1168) + ,Pad((1,0,1,0)) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = STARE(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/stare960all.py b/bob/ip/binseg/configs/datasets/stare960all.py new file mode 100644 index 0000000000000000000000000000000000000000..0b7f134c3110b49ca6f41820453e8e360a6b0d2b --- /dev/null +++ b/bob/ip/binseg/configs/datasets/stare960all.py @@ -0,0 +1,29 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +from bob.db.stare import Database as STARE +from bob.ip.binseg.data.transforms import * +from bob.ip.binseg.data.binsegdataset import BinSegDataset +import torch + +#### Config #### + +transforms = Compose([ + RandomRotation() + ,Pad((0,32,0,32)) + ,Resize(960) + ,CenterCrop(960) + ,RandomHFlip() + ,RandomVFlip() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = STARE(protocol = 'default') + +# PyTorch dataset +train = BinSegDataset(bobdb, split='train', transform=transforms) +test = BinSegDataset(bobdb, split='test', transform=transforms) + +dataset = torch.utils.data.ConcatDataset([train,test]) \ No newline at end of file