From 2d8475e9e04cf248948bacf7f6c7c8d63a33ceb8 Mon Sep 17 00:00:00 2001
From: Tim Laibacher <tim.laibacher@idiap.ch>
Date: Thu, 27 Jun 2019 10:27:31 +0200
Subject: [PATCH] Add new configs
---
.../configs/datasets/chasedb11024all.py | 28 +++++++++++++++
.../configs/datasets/chasedb11168all.py | 28 +++++++++++++++
.../binseg/configs/datasets/drive1024all.py | 28 +++++++++++++++
.../binseg/configs/datasets/drive1168all.py | 29 +++++++++++++++
bob/ip/binseg/configs/datasets/drive960all.py | 27 ++++++++++++++
.../drivestarechasedb1hrf1024ssliostar.py | 5 ++-
...rivestarechasedb1hrfl1024_all_ssliostar.py | 33 +++++++++++++++++
...drivestarechasedb1iostar1168_all_sslhrf.py | 35 +++++++++++++++++++
.../drivestarechasedb1iostar1168sslhrf.py | 6 ++--
.../drivestareiostarhrf960_all_sslchase.py | 35 +++++++++++++++++++
bob/ip/binseg/configs/datasets/hrf1024all.py | 28 +++++++++++++++
bob/ip/binseg/configs/datasets/hrf960all.py | 28 +++++++++++++++
.../configs/datasets/iostarvessel1168all.py | 29 +++++++++++++++
.../configs/datasets/iostarvessel960all.py | 27 ++++++++++++++
.../binseg/configs/datasets/stare1024all.py | 28 +++++++++++++++
.../binseg/configs/datasets/stare1168all.py | 28 +++++++++++++++
bob/ip/binseg/configs/datasets/stare960all.py | 29 +++++++++++++++
17 files changed, 445 insertions(+), 6 deletions(-)
create mode 100644 bob/ip/binseg/configs/datasets/chasedb11024all.py
create mode 100644 bob/ip/binseg/configs/datasets/chasedb11168all.py
create mode 100644 bob/ip/binseg/configs/datasets/drive1024all.py
create mode 100644 bob/ip/binseg/configs/datasets/drive1168all.py
create mode 100644 bob/ip/binseg/configs/datasets/drive960all.py
create mode 100644 bob/ip/binseg/configs/datasets/drivestarechasedb1hrfl1024_all_ssliostar.py
create mode 100644 bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168_all_sslhrf.py
create mode 100644 bob/ip/binseg/configs/datasets/drivestareiostarhrf960_all_sslchase.py
create mode 100644 bob/ip/binseg/configs/datasets/hrf1024all.py
create mode 100644 bob/ip/binseg/configs/datasets/hrf960all.py
create mode 100644 bob/ip/binseg/configs/datasets/iostarvessel1168all.py
create mode 100644 bob/ip/binseg/configs/datasets/iostarvessel960all.py
create mode 100644 bob/ip/binseg/configs/datasets/stare1024all.py
create mode 100644 bob/ip/binseg/configs/datasets/stare1168all.py
create mode 100644 bob/ip/binseg/configs/datasets/stare960all.py
diff --git a/bob/ip/binseg/configs/datasets/chasedb11024all.py b/bob/ip/binseg/configs/datasets/chasedb11024all.py
new file mode 100644
index 00000000..39d585cb
--- /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 00000000..e30c1ef7
--- /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 00000000..26d82bc8
--- /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 00000000..f1e9d856
--- /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 00000000..1242574b
--- /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 4dbd23d2..dd9016b7 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 00000000..7e68cb52
--- /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 00000000..fde74696
--- /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 376aae14..01705e15 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 00000000..5653591b
--- /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 00000000..c5fc4de2
--- /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 00000000..a34b3090
--- /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 00000000..b97d1ed0
--- /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 00000000..1338dc22
--- /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 00000000..46d80849
--- /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 00000000..d2384a09
--- /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 00000000..0b7f134c
--- /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
--
GitLab