diff --git a/bob/ip/binseg/configs/datasets/drivestareiostarhrf960.py b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960.py new file mode 100644 index 0000000000000000000000000000000000000000..911a718dcd6ff604e9acf2dc03fa3dfcf8ced254 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960.py @@ -0,0 +1,10 @@ +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 +import torch + +#### Config #### + +# PyTorch dataset +dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,iostar]) \ 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..94412e7059b24a8e6a3ebd71b4ce06a7d524ed64 --- /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) \ No newline at end of file diff --git a/setup.py b/setup.py index 1f45161c763c3afc916cef87c12886ee0a7dfc53..20671d7f66c740e8f36ddaf8e601020de7a8de4c 100644 --- a/setup.py +++ b/setup.py @@ -79,7 +79,9 @@ setup( 'DRIVE1168 = bob.ip.binseg.configs.datasets.drive1168', 'DRIVETEST = bob.ip.binseg.configs.datasets.drivetest', 'DRIVESTARECHASEDB1IOSTAR1168 = bob.ip.binseg.configs.datasets.drivestarechasedb1iostar1168', + 'DRIVESTAREIOSTARHRF960 = bob.ip.binseg.configs.datasets.drivestareiostarhrf960', 'HRF = bob.ip.binseg.configs.datasets.hrf', + 'HRF960 = bob.ip.binseg.configs.datasets.hrf960', 'HRF1168 = bob.ip.binseg.configs.datasets.hrf1168', 'HRFTEST = bob.ip.binseg.configs.datasets.hrftest', 'IOSTAROD = bob.ip.binseg.configs.datasets.iostarod',