Skip to content
Snippets Groups Projects
Commit b4465fef authored by Tim Laibacher's avatar Tim Laibacher
Browse files

Tidy up and add upscale configs

parent 8d9cf6c1
No related branches found
No related tags found
No related merge requests found
Pipeline #30971 failed
Stage: build
Stage: deploy
Hide whitespace changes
Inline Side-by-side
Showing
with 194 additions and 368 deletions
bob/ip/binseg/configs/datasets/allvessel544.py deleted 100644 → 0
+ 0
− 90
View file @ 8d9cf6c1
#!/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()]
# DRIVE
transforms_drive = Compose([
CenterCrop((544,544))
,*defaulttransforms
])
# bob.db.dataset init
bobdb_drive = DRIVE(protocol = 'default')
# PyTorch dataset
torch_drive = BinSegDataset(bobdb_drive, split='train', transform=transforms_drive)
# 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_drive,torch_stare, torch_chase, torch_iostar, torch_hrf])
\ No newline at end of file
bob/ip/binseg/configs/datasets/allvessel544test.py deleted 100644 → 0
+ 0
− 86
View file @ 8d9cf6c1
#!/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()]
# DRIVE
transforms_drive = Compose([
CenterCrop((544,544))
,*defaulttransforms
])
# bob.db.dataset init
bobdb_drive = DRIVE(protocol = 'default')
# PyTorch dataset
torch_drive = BinSegDataset(bobdb_drive, split='test', transform=transforms_drive)
# 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_drive,torch_stare, torch_chase, torch_iostar, torch_hrf])
\ No newline at end of file
bob/ip/binseg/configs/datasets/amdrive.py deleted 100644 → 0
+ 0
− 78
View file @ 8d9cf6c1
#!/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
bob/ip/binseg/configs/datasets/amdrivetest.py deleted 100644 → 0
+ 0
− 73
View file @ 8d9cf6c1
#!/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
bob/ip/binseg/configs/datasets/drive1024test.py → bob/ip/binseg/configs/datasets/chasedb11024.py
+ 24
− 0
View file @ b4465fef
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from bob.db.drive import Database as DRIVE
from bob.db.chasedb1 import Database as CHASEDB1
from bob.ip.binseg.data.transforms import *
from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
CenterCrop((540,540))
RandomRotation()
,Crop(0,18,960,960)
,Resize(1024)
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
# bob.db.dataset init
bobdb = DRIVE(protocol = 'default')
bobdb = CHASEDB1(protocol = 'default')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drive2336test.py → bob/ip/binseg/configs/datasets/chasedb11168.py
+ 24
− 0
View file @ b4465fef
#!/usr/bin/env python
# -*- coding: utf-8 -*-
from bob.db.drive import Database as DRIVE
from bob.db.chasedb1 import Database as CHASEDB1
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)
RandomRotation()
,Crop(140,18,680,960)
,Resize(1168)
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
# bob.db.dataset init
bobdb = DRIVE(protocol = 'default')
bobdb = CHASEDB1(protocol = 'default')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/chasedb1544test.py → bob/ip/binseg/configs/datasets/chasedb1544.py
+ 7
− 3
View file @ b4465fef
......@@ -7,9 +7,13 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Resize(544)
transforms = Compose([
RandomRotation()
,Resize(544)
,Crop(0,12,544,544)
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
......@@ -17,4 +21,4 @@ transforms = Compose([
bobdb = CHASEDB1(protocol = 'default')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drive1024.py
+ 2
− 2
View file @ b4465fef
......@@ -8,11 +8,11 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
CenterCrop((540,540))
RandomRotation()
,CenterCrop((540,540))
,Resize(1024)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
......
bob/ip/binseg/configs/datasets/drive2336.py → bob/ip/binseg/configs/datasets/drive1168.py
+ 3
− 3
View file @ b4465fef
......@@ -8,12 +8,12 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Crop(75,10,416,544)
RandomRotation()
,Crop(75,10,416,544)
,Pad((21,0,22,0))
,Resize(2336)
,Resize(1168)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
......
bob/ip/binseg/configs/datasets/drive608.py
+ 2
− 2
View file @ b4465fef
......@@ -8,12 +8,12 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
CenterCrop((470,544))
RandomRotation()
,CenterCrop((470,544))
,Pad((10,9,10,8))
,Resize(608)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
......
bob/ip/binseg/configs/datasets/drive608test.py deleted 100644 → 0
+ 0
− 21
View file @ 8d9cf6c1
#!/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
bob/ip/binseg/configs/datasets/drive960.py
+ 2
− 2
View file @ b4465fef
......@@ -8,11 +8,11 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
CenterCrop((544,544))
RandomRotation()
,CenterCrop((544,544))
,Resize(960)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
......
bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168.py 0 → 100644
+ 10
− 0
View file @ b4465fef
from bob.ip.binseg.configs.datasets.drive1168 import dataset as drive
from bob.ip.binseg.configs.datasets.stare1168 import dataset as stare
from bob.ip.binseg.configs.datasets.chasedb11168 import dataset as chase
from bob.ip.binseg.configs.datasets.iostarvessel1168 import dataset as iostar
import torch
#### Config ####
# PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,chase,iostar])
\ No newline at end of file
bob/ip/binseg/configs/datasets/hrf544test.py → bob/ip/binseg/configs/datasets/hrf1168.py
+ 7
− 3
View file @ b4465fef
......@@ -8,8 +8,12 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Resize((363))
,Pad((0,90,0,91))
Crop(0,108,2336,3296)
,Resize((1168))
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
......@@ -17,4 +21,4 @@ transforms = Compose([
bobdb = HRF(protocol = 'default')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/iostarvessel1168.py 0 → 100644
+ 25
− 0
View file @ b4465fef
#!/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
#### 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
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/iostarvessel544test.py → bob/ip/binseg/configs/datasets/iostarvessel960.py
+ 6
− 2
View file @ b4465fef
......@@ -8,7 +8,11 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Resize(544)
Resize(960)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
......@@ -16,4 +20,4 @@ transforms = Compose([
bobdb = IOSTAR(protocol='default_vessel')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/stare1024.py 0 → 100644
+ 25
− 0
View file @ b4465fef
#!/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
#### 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
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/stare1168.py 0 → 100644
+ 25
− 0
View file @ b4465fef
#!/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
#### 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
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/stare544test.py → bob/ip/binseg/configs/datasets/stare544.py
+ 7
− 3
View file @ b4465fef
......@@ -7,9 +7,13 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Resize(471)
transforms = Compose([ RandomRotation()
,Resize(471)
,Pad((0,37,0,36))
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
,ToTensor()
])
......@@ -17,4 +21,4 @@ transforms = Compose([
bobdb = STARE(protocol = 'default')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drive960test.py → bob/ip/binseg/configs/datasets/stare960.py
+ 25
− 0
View file @ b4465fef
#!/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.ip.binseg.data.transforms import *
from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
CenterCrop((544,544))
RandomRotation()
,Pad((0,32,0,32))
,Resize(960)
,CenterCrop(960)
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
# bob.db.dataset init
bobdb = DRIVE(protocol = 'default')
bobdb = STARE(protocol = 'default')
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment