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
Stage: build
Hide whitespace changes
Inline Side-by-side
Showing
with 196 additions and 165 deletions
bob/ip/binseg/configs/datasets/drive608.py
+ 14
− 12
View file @ 833335da
...@@ -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() [
,CenterCrop((470,544)) RandomRotation(),
,Pad((10,9,10,8)) CenterCrop((470, 544)),
,Resize(608) Pad((10, 9, 10, 8)),
,RandomHFlip() Resize(608),
,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
bob/ip/binseg/configs/datasets/drive960.py
+ 13
− 11
View file @ 833335da
...@@ -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((544,544)) RandomRotation(),
,Resize(960) CenterCrop((544, 544)),
,RandomHFlip() Resize(960),
,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
bob/ip/binseg/configs/datasets/drivechasedb1iostarhrf608.py
+ 1
− 1
View file @ 833335da
...@@ -7,4 +7,4 @@ import torch ...@@ -7,4 +7,4 @@ import torch
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,chase,iostar,hrf]) dataset = torch.utils.data.ConcatDataset([drive, chase, iostar, hrf])
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivechasedb1iostarhrf608sslstare.py
+ 21
− 13
View file @ 833335da
...@@ -5,30 +5,38 @@ from bob.ip.binseg.configs.datasets.hrf608 import dataset as hrf ...@@ -5,30 +5,38 @@ from bob.ip.binseg.configs.datasets.hrf608 import dataset as hrf
from bob.db.stare import Database as STARE from bob.db.stare import Database as STARE
from bob.ip.binseg.data.transforms import * from bob.ip.binseg.data.transforms import *
import torch import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
labeled_dataset = torch.utils.data.ConcatDataset([drive,chase,iostar,hrf]) labeled_dataset = torch.utils.data.ConcatDataset([drive, chase, iostar, hrf])
#### Unlabeled STARE TRAIN #### #### Unlabeled STARE TRAIN ####
unlabeled_transforms = Compose([ unlabeled_transforms = Compose(
Pad((2,1,2,2)) [
,RandomHFlip() Pad((2, 1, 2, 2)),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
starebobdb = STARE(protocol = 'default') starebobdb = STARE(protocol="default")
# PyTorch dataset # PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(starebobdb, split='train', transform=unlabeled_transforms) unlabeled_dataset = UnLabeledBinSegDataset(
starebobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset # SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivestarechasedb11168.py
+ 1
− 1
View file @ 833335da
...@@ -6,4 +6,4 @@ import torch ...@@ -6,4 +6,4 @@ import torch
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,chase]) dataset = torch.utils.data.ConcatDataset([drive, stare, chase])
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024.py
+ 1
− 1
View file @ 833335da
...@@ -7,4 +7,4 @@ import torch ...@@ -7,4 +7,4 @@ import torch
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,chase]) dataset = torch.utils.data.ConcatDataset([drive, stare, hrf, chase])
bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024ssliostar.py
+ 14
− 12
View file @ 833335da
...@@ -5,29 +5,31 @@ from bob.ip.binseg.configs.datasets.chasedb11024 import dataset as chasedb ...@@ -5,29 +5,31 @@ from bob.ip.binseg.configs.datasets.chasedb11024 import dataset as chasedb
from bob.db.iostar import Database as IOSTAR from bob.db.iostar import Database as IOSTAR
from bob.ip.binseg.data.transforms import * from bob.ip.binseg.data.transforms import *
import torch import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,chasedb]) labeled_dataset = torch.utils.data.ConcatDataset([drive, stare, hrf, chasedb])
#### Unlabeled IOSTAR Train #### #### Unlabeled IOSTAR Train ####
unlabeled_transforms = Compose([ unlabeled_transforms = Compose(
RandomHFlip() [RandomHFlip(), RandomVFlip(), RandomRotation(), ColorJitter(), ToTensor()]
,RandomVFlip() )
,RandomRotation()
,ColorJitter()
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
iostarbobdb = IOSTAR(protocol='default_vessel') iostarbobdb = IOSTAR(protocol="default_vessel")
# PyTorch dataset # PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(iostarbobdb, split='train', transform=unlabeled_transforms) unlabeled_dataset = UnLabeledBinSegDataset(
iostarbobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset # SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168.py
+ 1
− 1
View file @ 833335da
...@@ -7,4 +7,4 @@ import torch ...@@ -7,4 +7,4 @@ import torch
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,chase,iostar]) dataset = torch.utils.data.ConcatDataset([drive, stare, chase, iostar])
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168sslhrf.py
+ 22
− 14
View file @ 833335da
...@@ -5,31 +5,39 @@ from bob.ip.binseg.configs.datasets.iostarvessel1168 import dataset as iostar ...@@ -5,31 +5,39 @@ from bob.ip.binseg.configs.datasets.iostarvessel1168 import dataset as iostar
from bob.db.hrf import Database as HRF from bob.db.hrf import Database as HRF
from bob.ip.binseg.data.transforms import * from bob.ip.binseg.data.transforms import *
import torch import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,iostar,chasedb]) labeled_dataset = torch.utils.data.ConcatDataset([drive, stare, iostar, chasedb])
#### Unlabeled HRF TRAIN #### #### Unlabeled HRF TRAIN ####
unlabeled_transforms = Compose([ unlabeled_transforms = Compose(
RandomRotation() [
,Crop(0,108,2336,3296) RandomRotation(),
,Resize((1168)) Crop(0, 108, 2336, 3296),
,RandomHFlip() Resize((1168)),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
hrfbobdb = HRF(protocol='default') hrfbobdb = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(hrfbobdb, split='train', transform=unlabeled_transforms) unlabeled_dataset = UnLabeledBinSegDataset(
hrfbobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset # SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivestareiostarhrf960.py
+ 1
− 1
View file @ 833335da
...@@ -7,4 +7,4 @@ import torch ...@@ -7,4 +7,4 @@ import torch
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,iostar]) dataset = torch.utils.data.ConcatDataset([drive, stare, hrf, iostar])
bob/ip/binseg/configs/datasets/drivestareiostarhrf960sslchase.py
+ 21
− 13
View file @ 833335da
...@@ -6,30 +6,38 @@ from bob.db.chasedb1 import Database as CHASE ...@@ -6,30 +6,38 @@ from bob.db.chasedb1 import Database as CHASE
from bob.db.hrf import Database as HRF from bob.db.hrf import Database as HRF
from bob.ip.binseg.data.transforms import * from bob.ip.binseg.data.transforms import *
import torch import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config #### #### Config ####
# PyTorch dataset # PyTorch dataset
labeled_dataset = torch.utils.data.ConcatDataset([drive,stare,hrf,iostar]) labeled_dataset = torch.utils.data.ConcatDataset([drive, stare, hrf, iostar])
#### Unlabeled CHASE TRAIN #### #### Unlabeled CHASE TRAIN ####
unlabeled_transforms = Compose([ unlabeled_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
chasebobdb = CHASE(protocol = 'default') chasebobdb = CHASE(protocol="default")
# PyTorch dataset # PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(chasebobdb, split='train', transform=unlabeled_transforms) unlabeled_dataset = UnLabeledBinSegDataset(
chasebobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset # SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset) dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
bob/ip/binseg/configs/datasets/drivetest.py
+ 3
− 6
View file @ 833335da
...@@ -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((544, 544)), ToTensor()])
CenterCrop((544,544))
,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
bob/ip/binseg/configs/datasets/hrf.py
+ 12
− 10
View file @ 833335da
...@@ -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,108,2336,3296) [
,RandomHFlip() Crop(0, 108, 2336, 3296),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(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
bob/ip/binseg/configs/datasets/hrf1024.py
+ 13
− 11
View file @ 833335da
...@@ -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(
Pad((0,584,0,584)) [
,Resize((1024)) Pad((0, 584, 0, 584)),
,RandomRotation() Resize((1024)),
,RandomHFlip() RandomRotation(),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/hrf1168.py
+ 13
− 11
View file @ 833335da
...@@ -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(
Crop(0,108,2336,3296) [
,Resize((1168)) Crop(0, 108, 2336, 3296),
,RandomHFlip() Resize((1168)),
,RandomVFlip() RandomHFlip(),
,RandomRotation() RandomVFlip(),
,ColorJitter() RandomRotation(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(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
bob/ip/binseg/configs/datasets/hrf1168test.py
+ 3
− 7
View file @ 833335da
...@@ -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([Crop(0, 108, 2336, 3296), Resize((1168)), ToTensor()])
Crop(0,108,2336,3296)
,Resize((1168))
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(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
bob/ip/binseg/configs/datasets/hrf544.py
+ 13
− 11
View file @ 833335da
...@@ -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((363)) [
,Pad((0,90,0,91)) Resize((363)),
,RandomRotation() Pad((0, 90, 0, 91)),
,RandomHFlip() RandomRotation(),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/hrf544test.py
+ 3
− 7
View file @ 833335da
...@@ -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([Resize((363)), Pad((0, 90, 0, 91)), ToTensor()])
Resize((363))
,Pad((0,90,0,91))
,ToTensor()
])
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(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
bob/ip/binseg/configs/datasets/hrf608.py
+ 13
− 11
View file @ 833335da
...@@ -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(
Pad((0,345,0,345)) [
,Resize(608) Pad((0, 345, 0, 345)),
,RandomRotation() Resize(608),
,RandomHFlip() RandomRotation(),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/hrf960.py
+ 13
− 11
View file @ 833335da
...@@ -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(
Pad((0,584,0,584)) [
,Resize((960)) Pad((0, 584, 0, 584)),
,RandomRotation() Resize((960)),
,RandomHFlip() RandomRotation(),
,RandomVFlip() RandomHFlip(),
,ColorJitter() RandomVFlip(),
,ToTensor() ColorJitter(),
]) ToTensor(),
]
)
# bob.db.dataset init # bob.db.dataset init
bobdb = HRF(protocol = 'default') bobdb = HRF(protocol="default")
# PyTorch dataset # PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms) dataset = BinSegDataset(bobdb, split="train", transform=transforms)
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