diff --git a/bob/ip/binseg/configs/datasets/amdrive.py b/bob/ip/binseg/configs/datasets/amdrive.py new file mode 100644 index 0000000000000000000000000000000000000000..6f7cc3101d158f71c7ea47f32382ce369de0552b --- /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 0000000000000000000000000000000000000000..026ac236e7e9f48f5f41b6685d01369606c47321 --- /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 0000000000000000000000000000000000000000..d5f08ff8742d49652de783183d44b8e3eec1b3a3 --- /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 0000000000000000000000000000000000000000..9e9cb3e935a63ddb45bb0bba85c56c173de6912d --- /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 0000000000000000000000000000000000000000..30ec9af8400daa56bece1318ca8a07a56254a8c4 --- /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 0000000000000000000000000000000000000000..9ce3b6491c47379abe09886bd9d904afde1ccc00 --- /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 0000000000000000000000000000000000000000..64037a07feeef8cff2c85b1a787eb0bafd1cd168 --- /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 0000000000000000000000000000000000000000..7c59713624bbfab1f95fd5f4b079951258e40a99 --- /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 0000000000000000000000000000000000000000..f0d06b781a7135f929ae28a53b9c83ea268c3f20 --- /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 0000000000000000000000000000000000000000..041fa775f9ce6a47c6f8db145fed22d009e86cde --- /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 1bc10ef053de55522058198ea36267c4b8c60ec5..0e8fa1328167de67c1ca55f27791e9b2eefd6efd 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 199a249a59f25c13d6f987c3f784822d905887f7..7ad9bb836760b19c021f05d43ea9886685ce3257 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 306ecb80fecb8cbc17623aad22efd48375798397..eeb0e599e89c967247975456016dc35facdf805c 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 cb422dca122bda790376589f2989acebca76dd6f..b15a277966200c97b47a201ff086c7002b9655aa 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 1a049f14cc813a465ade9859ed7c17799e258e24..a1db473cc7f87b6847f57cc1200326d42cb8ab86 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 7c21b042dc0b1c4c76a78fa5c7a93a8d5d5b2eee..8182c7fa089c77221b4434e272ae6e0d394b3912 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