diff --git a/bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py new file mode 100644 index 0000000000000000000000000000000000000000..a7bd4576766259e83ec633bfac15b4dc46f0da9d --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py @@ -0,0 +1,35 @@ +from bob.ip.binseg.configs.datasets.drive960 import dataset as drive +from bob.ip.binseg.configs.datasets.stare960 import dataset as stare +from bob.ip.binseg.configs.datasets.hrf960 import dataset as hrf +from bob.ip.binseg.configs.datasets.iostarvessel960 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/hrf960.py b/bob/ip/binseg/configs/datasets/hrf960.py new file mode 100644 index 0000000000000000000000000000000000000000..dd43cf00b68e8a66e8e0ec362ec519bf3488358e --- /dev/null +++ b/bob/ip/binseg/configs/datasets/hrf960.py @@ -0,0 +1,24 @@ +#!/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 + +#### 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 +dataset = BinSegDataset(bobdb, split='train', transform=transforms)