From 18dc9c0a3b431bc24c34e08bae19051f0e896e9b Mon Sep 17 00:00:00 2001 From: Tim Laibacher <tim.laibacher@idiap.ch> Date: Thu, 23 May 2019 08:43:01 +0200 Subject: [PATCH] Update configs --- bob/ip/binseg/configs/datasets/amdrive.py | 78 +++++++++++++++++++ bob/ip/binseg/configs/datasets/amdrivetest.py | 73 +++++++++++++++++ bob/ip/binseg/configs/datasets/drive1024.py | 24 ++++++ .../binseg/configs/datasets/drive1024test.py | 20 +++++ bob/ip/binseg/configs/datasets/drive2336.py | 25 ++++++ .../binseg/configs/datasets/drive2336test.py | 21 +++++ bob/ip/binseg/configs/datasets/drive608.py | 25 ++++++ .../binseg/configs/datasets/drive608test.py | 21 +++++ bob/ip/binseg/configs/datasets/drive960.py | 24 ++++++ .../binseg/configs/datasets/drive960test.py | 20 +++++ bob/ip/binseg/configs/models/driu.py | 2 +- bob/ip/binseg/configs/models/driuod.py | 2 +- bob/ip/binseg/configs/models/hed.py | 2 +- bob/ip/binseg/configs/models/m2unet.py | 2 +- bob/ip/binseg/configs/models/resunet.py | 2 +- bob/ip/binseg/configs/models/unet.py | 2 +- 16 files changed, 337 insertions(+), 6 deletions(-) create mode 100644 bob/ip/binseg/configs/datasets/amdrive.py create mode 100644 bob/ip/binseg/configs/datasets/amdrivetest.py create mode 100644 bob/ip/binseg/configs/datasets/drive1024.py create mode 100644 bob/ip/binseg/configs/datasets/drive1024test.py create mode 100644 bob/ip/binseg/configs/datasets/drive2336.py create mode 100644 bob/ip/binseg/configs/datasets/drive2336test.py create mode 100644 bob/ip/binseg/configs/datasets/drive608.py create mode 100644 bob/ip/binseg/configs/datasets/drive608test.py create mode 100644 bob/ip/binseg/configs/datasets/drive960.py create mode 100644 bob/ip/binseg/configs/datasets/drive960test.py diff --git a/bob/ip/binseg/configs/datasets/amdrive.py b/bob/ip/binseg/configs/datasets/amdrive.py new file mode 100644 index 00000000..6f7cc310 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/amdrive.py @@ -0,0 +1,78 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +from bob.db.drive import Database as DRIVE +from bob.db.stare import Database as STARE +from bob.db.chasedb1 import Database as CHASEDB1 +from bob.db.iostar import Database as IOSTAR +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 + +# Target size: 544x544 (DRIVE) + +defaulttransforms = [RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor()] + + + +# CHASE_DB1 +transforms_chase = Compose([ + Resize(544) + ,Crop(0,12,544,544) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_chase = CHASEDB1(protocol = 'default') + +# PyTorch dataset +torch_chase = BinSegDataset(bobdb_chase, split='train', transform=transforms_chase) + + +# IOSTAR VESSEL +transforms_iostar = Compose([ + Resize(544) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_iostar = IOSTAR(protocol='default_vessel') + +# PyTorch dataset +torch_iostar = BinSegDataset(bobdb_iostar, split='train', transform=transforms_iostar) + +# STARE +transforms = Compose([ + Resize(471) + ,Pad((0,37,0,36)) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_stare = STARE(protocol = 'default') + +# PyTorch dataset +torch_stare = BinSegDataset(bobdb_stare, split='train', transform=transforms) + + +# HRF +transforms_hrf = Compose([ + Resize((363)) + ,Pad((0,90,0,91)) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_hrf = HRF(protocol = 'default') + +# PyTorch dataset +torch_hrf = BinSegDataset(bobdb_hrf, split='train', transform=transforms_hrf) + + + +# Merge +dataset = torch.utils.data.ConcatDataset([torch_stare, torch_chase, torch_iostar, torch_hrf]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/amdrivetest.py b/bob/ip/binseg/configs/datasets/amdrivetest.py new file mode 100644 index 00000000..026ac236 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/amdrivetest.py @@ -0,0 +1,73 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +from bob.db.drive import Database as DRIVE +from bob.db.stare import Database as STARE +from bob.db.chasedb1 import Database as CHASEDB1 +from bob.db.iostar import Database as IOSTAR +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 + +# Target size: 544x544 (DRIVE) + +defaulttransforms = [ToTensor()] + + +# CHASE_DB1 +transforms_chase = Compose([ + Resize(544) + ,Crop(0,12,544,544) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_chase = CHASEDB1(protocol = 'default') + +# PyTorch dataset +torch_chase = BinSegDataset(bobdb_chase, split='test', transform=transforms_chase) + + +# IOSTAR VESSEL +transforms_iostar = Compose([ + Resize(544) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_iostar = IOSTAR(protocol='default_vessel') + +# PyTorch dataset +torch_iostar = BinSegDataset(bobdb_iostar, split='test', transform=transforms_iostar) + +# STARE +transforms = Compose([ + Resize(471) + ,Pad((0,37,0,36)) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_stare = STARE(protocol = 'default') + +# PyTorch dataset +torch_stare = BinSegDataset(bobdb_stare, split='test', transform=transforms) + + +# HRF +transforms_hrf = Compose([ + Resize((363)) + ,Pad((0,90,0,91)) + ,*defaulttransforms + ]) + +# bob.db.dataset init +bobdb_hrf = HRF(protocol = 'default') + +# PyTorch dataset +torch_hrf = BinSegDataset(bobdb_hrf, split='test', transform=transforms_hrf) + + + +# Merge +dataset = torch.utils.data.ConcatDataset([torch_stare, torch_chase, torch_iostar, torch_hrf]) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive1024.py b/bob/ip/binseg/configs/datasets/drive1024.py new file mode 100644 index 00000000..d5f08ff8 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive1024.py @@ -0,0 +1,24 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + CenterCrop((540,540)) + ,Resize(1024) + ,RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='train', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive1024test.py b/bob/ip/binseg/configs/datasets/drive1024test.py new file mode 100644 index 00000000..9e9cb3e9 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive1024test.py @@ -0,0 +1,20 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + CenterCrop((540,540)) + ,Resize(1024) + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='test', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive2336.py b/bob/ip/binseg/configs/datasets/drive2336.py new file mode 100644 index 00000000..30ec9af8 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive2336.py @@ -0,0 +1,25 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + Crop(75,10,416,544) + ,Pad((21,0,22,0)) + ,Resize(2336) + ,RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='train', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive2336test.py b/bob/ip/binseg/configs/datasets/drive2336test.py new file mode 100644 index 00000000..9ce3b649 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive2336test.py @@ -0,0 +1,21 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + Crop(75,10,416,544) + ,Pad((21,0,22,0)) + ,Resize(2336) + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='test', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive608.py b/bob/ip/binseg/configs/datasets/drive608.py new file mode 100644 index 00000000..64037a07 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive608.py @@ -0,0 +1,25 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + CenterCrop((470,544)) + ,Pad((10,9,10,8)) + ,Resize(608) + ,RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='train', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive608test.py b/bob/ip/binseg/configs/datasets/drive608test.py new file mode 100644 index 00000000..7c597136 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive608test.py @@ -0,0 +1,21 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + CenterCrop((470,544)) + ,Pad((10,9,10,8)) + ,Resize(608) + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='test', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive960.py b/bob/ip/binseg/configs/datasets/drive960.py new file mode 100644 index 00000000..f0d06b78 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive960.py @@ -0,0 +1,24 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + CenterCrop((544,544)) + ,Resize(960) + ,RandomHFlip() + ,RandomVFlip() + ,RandomRotation() + ,ColorJitter() + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='train', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/datasets/drive960test.py b/bob/ip/binseg/configs/datasets/drive960test.py new file mode 100644 index 00000000..041fa775 --- /dev/null +++ b/bob/ip/binseg/configs/datasets/drive960test.py @@ -0,0 +1,20 @@ +#!/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 + +#### Config #### + +transforms = Compose([ + CenterCrop((544,544)) + ,Resize(960) + ,ToTensor() + ]) + +# bob.db.dataset init +bobdb = DRIVE(protocol = 'default') + +# PyTorch dataset +dataset = BinSegDataset(bobdb, split='test', transform=transforms) \ No newline at end of file diff --git a/bob/ip/binseg/configs/models/driu.py b/bob/ip/binseg/configs/models/driu.py index 1bc10ef0..0e8fa132 100644 --- a/bob/ip/binseg/configs/models/driu.py +++ b/bob/ip/binseg/configs/models/driu.py @@ -19,7 +19,7 @@ gamma = 1e-3 eps = 1e-8 amsbound = False -scheduler_milestones = [800] +scheduler_milestones = [900] scheduler_gamma = 0.1 # model diff --git a/bob/ip/binseg/configs/models/driuod.py b/bob/ip/binseg/configs/models/driuod.py index 199a249a..7ad9bb83 100644 --- a/bob/ip/binseg/configs/models/driuod.py +++ b/bob/ip/binseg/configs/models/driuod.py @@ -19,7 +19,7 @@ gamma = 1e-3 eps = 1e-8 amsbound = False -scheduler_milestones = [800] +scheduler_milestones = [900] scheduler_gamma = 0.1 # model diff --git a/bob/ip/binseg/configs/models/hed.py b/bob/ip/binseg/configs/models/hed.py index 306ecb80..eeb0e599 100644 --- a/bob/ip/binseg/configs/models/hed.py +++ b/bob/ip/binseg/configs/models/hed.py @@ -19,7 +19,7 @@ gamma = 1e-3 eps = 1e-8 amsbound = False -scheduler_milestones = [800] +scheduler_milestones = [900] scheduler_gamma = 0.1 diff --git a/bob/ip/binseg/configs/models/m2unet.py b/bob/ip/binseg/configs/models/m2unet.py index cb422dca..b15a2779 100644 --- a/bob/ip/binseg/configs/models/m2unet.py +++ b/bob/ip/binseg/configs/models/m2unet.py @@ -19,7 +19,7 @@ gamma = 1e-3 eps = 1e-8 amsbound = False -scheduler_milestones = [800] +scheduler_milestones = [900] scheduler_gamma = 0.1 # model diff --git a/bob/ip/binseg/configs/models/resunet.py b/bob/ip/binseg/configs/models/resunet.py index 1a049f14..a1db473c 100644 --- a/bob/ip/binseg/configs/models/resunet.py +++ b/bob/ip/binseg/configs/models/resunet.py @@ -19,7 +19,7 @@ gamma = 1e-3 eps = 1e-8 amsbound = False -scheduler_milestones = [800] +scheduler_milestones = [900] scheduler_gamma = 0.1 # model diff --git a/bob/ip/binseg/configs/models/unet.py b/bob/ip/binseg/configs/models/unet.py index 7c21b042..8182c7fa 100644 --- a/bob/ip/binseg/configs/models/unet.py +++ b/bob/ip/binseg/configs/models/unet.py @@ -19,7 +19,7 @@ gamma = 1e-3 eps = 1e-8 amsbound = False -scheduler_milestones = [800] +scheduler_milestones = [900] scheduler_gamma = 0.1 # model -- GitLab