From 40f7f4192ee2fcde49d54320c0c9558cf99b5921 Mon Sep 17 00:00:00 2001
From: Tim Laibacher <tim.laibacher@idiap.ch>
Date: Mon, 24 Jun 2019 17:22:06 +0200
Subject: [PATCH] Add configs

---
 .../drivestareiostarhrf960sslchase.py         | 35 +++++++++++++++++++
 bob/ip/binseg/configs/datasets/hrf960.py      | 24 +++++++++++++
 2 files changed, 59 insertions(+)
 create mode 100644 bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py
 create mode 100644 bob/ip/binseg/configs/datasets/hrf960.py

diff --git a/bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py b/bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py
new file mode 100644
index 00000000..a7bd4576
--- /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 00000000..dd43cf00
--- /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)
-- 
GitLab