Add more configs

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

+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

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

+#!/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

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',