Skip to content
Snippets Groups Projects
Commit 833335da authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[all] Passed black on all python files

parent c3dc9915
No related branches found
No related tags found
1 merge request!12Streamlining
Pipeline #38202 passed
Showing
with 196 additions and 215 deletions
# see https://docs.python.org/3/library/pkgutil.html # see https://docs.python.org/3/library/pkgutil.html
from pkgutil import extend_path from pkgutil import extend_path
__path__ = extend_path(__path__, __name__) __path__ = extend_path(__path__, __name__)
# see https://docs.python.org/3/library/pkgutil.html # see https://docs.python.org/3/library/pkgutil.html
from pkgutil import extend_path from pkgutil import extend_path
__path__ = extend_path(__path__, __name__)
\ No newline at end of file __path__ = extend_path(__path__, __name__)
...@@ -11,68 +11,55 @@ import torch ...@@ -11,68 +11,55 @@ import torch
# Target size: 544x544 (DRIVE) # Target size: 544x544 (DRIVE)
defaulttransforms = [RandomHFlip() defaulttransforms = [
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor()] ColorJitter(),
ToTensor(),
]
# CHASE_DB1
# CHASE_DB1 transforms_chase = Compose([Resize(544), Crop(0, 12, 544, 544), *defaulttransforms])
transforms_chase = Compose([
Resize(544)
,Crop(0,12,544,544)
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_chase = CHASEDB1(protocol = 'default') bobdb_chase = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
torch_chase = BinSegDataset(bobdb_chase, split='train', transform=transforms_chase) torch_chase = BinSegDataset(bobdb_chase, split="train", transform=transforms_chase)
# IOSTAR VESSEL # IOSTAR VESSEL
transforms_iostar = Compose([ transforms_iostar = Compose([Resize(544), *defaulttransforms])
Resize(544)
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_iostar = IOSTAR(protocol='default_vessel') bobdb_iostar = IOSTAR(protocol="default_vessel")
# PyTorch dataset # PyTorch dataset
torch_iostar = BinSegDataset(bobdb_iostar, split='train', transform=transforms_iostar) torch_iostar = BinSegDataset(bobdb_iostar, split="train", transform=transforms_iostar)
# STARE # STARE
transforms = Compose([ transforms = Compose([Resize(471), Pad((0, 37, 0, 36)), *defaulttransforms])
Resize(471)
,Pad((0,37,0,36))
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_stare = STARE(protocol = 'default') bobdb_stare = STARE(protocol="default")
# PyTorch dataset # PyTorch dataset
torch_stare = BinSegDataset(bobdb_stare, split='train', transform=transforms) torch_stare = BinSegDataset(bobdb_stare, split="train", transform=transforms)
# HRF # HRF
transforms_hrf = Compose([ transforms_hrf = Compose([Resize((363)), Pad((0, 90, 0, 91)), *defaulttransforms])
Resize((363))
,Pad((0,90,0,91))
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_hrf = HRF(protocol = 'default') bobdb_hrf = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
torch_hrf = BinSegDataset(bobdb_hrf, split='train', transform=transforms_hrf) torch_hrf = BinSegDataset(bobdb_hrf, split="train", transform=transforms_hrf)
# Merge # Merge
dataset = torch.utils.data.ConcatDataset([torch_stare, torch_chase, torch_iostar, torch_hrf]) dataset = torch.utils.data.ConcatDataset(
\ No newline at end of file [torch_stare, torch_chase, torch_iostar, torch_hrf]
)
...@@ -14,60 +14,46 @@ import torch ...@@ -14,60 +14,46 @@ import torch
defaulttransforms = [ToTensor()] defaulttransforms = [ToTensor()]
# CHASE_DB1 # CHASE_DB1
transforms_chase = Compose([ transforms_chase = Compose([Resize(544), Crop(0, 12, 544, 544), *defaulttransforms])
Resize(544)
,Crop(0,12,544,544)
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_chase = CHASEDB1(protocol = 'default') bobdb_chase = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
torch_chase = BinSegDataset(bobdb_chase, split='test', transform=transforms_chase) torch_chase = BinSegDataset(bobdb_chase, split="test", transform=transforms_chase)
# IOSTAR VESSEL # IOSTAR VESSEL
transforms_iostar = Compose([ transforms_iostar = Compose([Resize(544), *defaulttransforms])
Resize(544)
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_iostar = IOSTAR(protocol='default_vessel') bobdb_iostar = IOSTAR(protocol="default_vessel")
# PyTorch dataset # PyTorch dataset
torch_iostar = BinSegDataset(bobdb_iostar, split='test', transform=transforms_iostar) torch_iostar = BinSegDataset(bobdb_iostar, split="test", transform=transforms_iostar)
# STARE # STARE
transforms = Compose([ transforms = Compose([Resize(471), Pad((0, 37, 0, 36)), *defaulttransforms])
Resize(471)
,Pad((0,37,0,36))
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_stare = STARE(protocol = 'default') bobdb_stare = STARE(protocol="default")
# PyTorch dataset # PyTorch dataset
torch_stare = BinSegDataset(bobdb_stare, split='test', transform=transforms) torch_stare = BinSegDataset(bobdb_stare, split="test", transform=transforms)
# HRF # HRF
transforms_hrf = Compose([ transforms_hrf = Compose([Resize((363)), Pad((0, 90, 0, 91)), *defaulttransforms])
Resize((363))
,Pad((0,90,0,91))
,*defaulttransforms
])
# bob.db.dataset init # bob.db.dataset init
bobdb_hrf = HRF(protocol = 'default') bobdb_hrf = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
torch_hrf = BinSegDataset(bobdb_hrf, split='test', transform=transforms_hrf) torch_hrf = BinSegDataset(bobdb_hrf, split="test", transform=transforms_hrf)
# Merge # Merge
dataset = torch.utils.data.ConcatDataset([torch_stare, torch_chase, torch_iostar, torch_hrf]) dataset = torch.utils.data.ConcatDataset(
\ No newline at end of file [torch_stare, torch_chase, torch_iostar, torch_hrf]
)
...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
Crop(0,18,960,960) [
,RandomHFlip() Crop(0, 18, 960, 960),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = CHASEDB1(protocol = 'default') bobdb = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
RandomRotation() [
,Crop(0,18,960,960) RandomRotation(),
,Resize(1024) Crop(0, 18, 960, 960),
,RandomHFlip() Resize(1024),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = CHASEDB1(protocol = 'default') bobdb = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
RandomRotation() [
,Crop(140,18,680,960) RandomRotation(),
,Resize(1168) Crop(140, 18, 680, 960),
,RandomHFlip() Resize(1168),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = CHASEDB1(protocol = 'default') bobdb = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
Resize(544) [
,Crop(0,12,544,544) Resize(544),
,RandomHFlip() Crop(0, 12, 544, 544),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = CHASEDB1(protocol = 'default') bobdb = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
RandomRotation() [
,CenterCrop((829,960)) RandomRotation(),
,Resize(608) CenterCrop((829, 960)),
,RandomHFlip() Resize(608),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = CHASEDB1(protocol = 'default') bobdb = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose([Crop(0, 18, 960, 960), ToTensor()])
Crop(0,18,960,960)
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = CHASEDB1(protocol = 'default') bobdb = CHASEDB1(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms) dataset = BinSegDataset(bobdb, split="test", transform=transforms)
\ No newline at end of file
...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
Pad((4,8,4,8)) [
,RandomHFlip() Pad((4, 8, 4, 8)),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = DRIONS(protocol = 'default') bobdb = DRIONS(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose([Pad((4, 8, 4, 8)), ToTensor()])
Pad((4,8,4,8))
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = DRIONS(protocol = 'default') bobdb = DRIONS(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms) dataset = BinSegDataset(bobdb, split="test", transform=transforms)
\ No newline at end of file
...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
CenterCrop((1760,2048)) [
,RandomHFlip() CenterCrop((1760, 2048)),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = DRISHTI(protocol = 'default_cup') bobdb = DRISHTI(protocol="default_cup")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -6,13 +6,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -6,13 +6,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose([CenterCrop((1760, 2048)), ToTensor()])
CenterCrop((1760,2048))
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = DRISHTI(protocol = 'default_cup') bobdb = DRISHTI(protocol="default_cup")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms) dataset = BinSegDataset(bobdb, split="test", transform=transforms)
\ No newline at end of file
...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
CenterCrop((1760,2048)) [
,RandomHFlip() CenterCrop((1760, 2048)),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = DRISHTI(protocol = 'default_od') bobdb = DRISHTI(protocol="default_od")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose([CenterCrop((1760, 2048)), ToTensor()])
CenterCrop((1760,2048))
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = DRISHTI(protocol = 'default_od') bobdb = DRISHTI(protocol="default_od")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms) dataset = BinSegDataset(bobdb, split="test", transform=transforms)
\ No newline at end of file
...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
CenterCrop((544,544)) [
,RandomHFlip() CenterCrop((544, 544)),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = DRIVE(protocol = 'default') bobdb = DRIVE(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
RandomRotation() [
,CenterCrop((540,540)) RandomRotation(),
,Resize(1024) CenterCrop((540, 540)),
,RandomHFlip() Resize(1024),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = DRIVE(protocol = 'default') bobdb = DRIVE(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
...@@ -7,14 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,14 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose([CenterCrop((540, 540)), Resize(1024), ToTensor()])
CenterCrop((540,540))
,Resize(1024)
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = DRIVE(protocol = 'default') bobdb = DRIVE(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms) dataset = BinSegDataset(bobdb, split="test", transform=transforms)
\ No newline at end of file
...@@ -7,19 +7,21 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset ...@@ -7,19 +7,21 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config #### #### Config ####
transforms = Compose([ transforms = Compose(
RandomRotation() [
,Crop(75,10,416,544) RandomRotation(),
,Pad((21,0,22,0)) Crop(75, 10, 416, 544),
,Resize(1168) Pad((21, 0, 22, 0)),
,RandomHFlip() Resize(1168),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = DRIVE(protocol = 'default') bobdb = DRIVE(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment