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
#### Config ####
transforms = Compose([
RandomRotation()
,CenterCrop((470,544))
,Pad((10,9,10,8))
,Resize(608)
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
RandomRotation(),
CenterCrop((470, 544)),
Pad((10, 9, 10, 8)),
Resize(608),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = DRIVE(protocol = 'default')
bobdb = DRIVE(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/drive960.py
+ 13
− 11
View file @ 833335da
......@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
RandomRotation()
,CenterCrop((544,544))
,Resize(960)
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
RandomRotation(),
CenterCrop((544, 544)),
Resize(960),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = DRIVE(protocol = 'default')
bobdb = DRIVE(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/drivechasedb1iostarhrf608.py
+ 1
− 1
View file @ 833335da
......@@ -7,4 +7,4 @@ import torch
#### Config ####
# PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,chase,iostar,hrf])
\ No newline at end of file
dataset = torch.utils.data.ConcatDataset([drive, chase, iostar, hrf])
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
from bob.db.stare import Database as STARE
from bob.ip.binseg.data.transforms import *
import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset
from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config ####
# 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_transforms = Compose([
Pad((2,1,2,2))
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
unlabeled_transforms = Compose(
[
Pad((2, 1, 2, 2)),
RandomHFlip(),
RandomVFlip(),
RandomRotation(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
starebobdb = STARE(protocol = 'default')
starebobdb = STARE(protocol="default")
# PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(starebobdb, split='train', transform=unlabeled_transforms)
unlabeled_dataset = UnLabeledBinSegDataset(
starebobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
bob/ip/binseg/configs/datasets/drivestarechasedb11168.py
+ 1
− 1
View file @ 833335da
......@@ -6,4 +6,4 @@ import torch
#### Config ####
# PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,chase])
\ No newline at end of file
dataset = torch.utils.data.ConcatDataset([drive, stare, chase])
bob/ip/binseg/configs/datasets/drivestarechasedb1hrf1024.py
+ 1
− 1
View file @ 833335da
......@@ -7,4 +7,4 @@ import torch
#### Config ####
# 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
from bob.db.iostar import Database as IOSTAR
from bob.ip.binseg.data.transforms import *
import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset
from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config ####
# 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_transforms = Compose([
RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
unlabeled_transforms = Compose(
[RandomHFlip(), RandomVFlip(), RandomRotation(), ColorJitter(), ToTensor()]
)
# bob.db.dataset init
iostarbobdb = IOSTAR(protocol='default_vessel')
iostarbobdb = IOSTAR(protocol="default_vessel")
# PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(iostarbobdb, split='train', transform=unlabeled_transforms)
unlabeled_dataset = UnLabeledBinSegDataset(
iostarbobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
bob/ip/binseg/configs/datasets/drivestarechasedb1iostar1168.py
+ 1
− 1
View file @ 833335da
......@@ -7,4 +7,4 @@ import torch
#### Config ####
# PyTorch dataset
dataset = torch.utils.data.ConcatDataset([drive,stare,chase,iostar])
\ No newline at end of file
dataset = torch.utils.data.ConcatDataset([drive, stare, chase, iostar])
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
from bob.db.hrf import Database as HRF
from bob.ip.binseg.data.transforms import *
import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset
from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config ####
# 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_transforms = Compose([
RandomRotation()
,Crop(0,108,2336,3296)
,Resize((1168))
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
unlabeled_transforms = Compose(
[
RandomRotation(),
Crop(0, 108, 2336, 3296),
Resize((1168)),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
hrfbobdb = HRF(protocol='default')
hrfbobdb = HRF(protocol="default")
# PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(hrfbobdb, split='train', transform=unlabeled_transforms)
unlabeled_dataset = UnLabeledBinSegDataset(
hrfbobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
bob/ip/binseg/configs/datasets/drivestareiostarhrf960.py
+ 1
− 1
View file @ 833335da
......@@ -7,4 +7,4 @@ import torch
#### Config ####
# 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
from bob.db.hrf import Database as HRF
from bob.ip.binseg.data.transforms import *
import torch
from bob.ip.binseg.data.binsegdataset import BinSegDataset, SSLBinSegDataset, UnLabeledBinSegDataset
from bob.ip.binseg.data.binsegdataset import (
BinSegDataset,
SSLBinSegDataset,
UnLabeledBinSegDataset,
)
#### Config ####
# 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_transforms = Compose([
Crop(0,18,960,960)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
unlabeled_transforms = Compose(
[
Crop(0, 18, 960, 960),
RandomHFlip(),
RandomVFlip(),
RandomRotation(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
chasebobdb = CHASE(protocol = 'default')
chasebobdb = CHASE(protocol="default")
# PyTorch dataset
unlabeled_dataset = UnLabeledBinSegDataset(chasebobdb, split='train', transform=unlabeled_transforms)
unlabeled_dataset = UnLabeledBinSegDataset(
chasebobdb, split="train", transform=unlabeled_transforms
)
# SSL Dataset
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
\ No newline at end of file
dataset = SSLBinSegDataset(labeled_dataset, unlabeled_dataset)
bob/ip/binseg/configs/datasets/drivetest.py
+ 3
− 6
View file @ 833335da
......@@ -7,13 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
CenterCrop((544,544))
,ToTensor()
])
transforms = Compose([CenterCrop((544, 544)), ToTensor()])
# bob.db.dataset init
bobdb = DRIVE(protocol = 'default')
bobdb = DRIVE(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="test", transform=transforms)
bob/ip/binseg/configs/datasets/hrf.py
+ 12
− 10
View file @ 833335da
......@@ -7,17 +7,19 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Crop(0,108,2336,3296)
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
Crop(0, 108, 2336, 3296),
RandomHFlip(),
RandomVFlip(),
RandomRotation(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/hrf1024.py
+ 13
− 11
View file @ 833335da
......@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Pad((0,584,0,584))
,Resize((1024))
,RandomRotation()
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
Pad((0, 584, 0, 584)),
Resize((1024)),
RandomRotation(),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# 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
#### Config ####
transforms = Compose([
Crop(0,108,2336,3296)
,Resize((1168))
,RandomHFlip()
,RandomVFlip()
,RandomRotation()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
Crop(0, 108, 2336, 3296),
Resize((1168)),
RandomHFlip(),
RandomVFlip(),
RandomRotation(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='train', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="train", transform=transforms)
bob/ip/binseg/configs/datasets/hrf1168test.py
+ 3
− 7
View file @ 833335da
......@@ -7,14 +7,10 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Crop(0,108,2336,3296)
,Resize((1168))
,ToTensor()
])
transforms = Compose([Crop(0, 108, 2336, 3296), Resize((1168)), ToTensor()])
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="test", transform=transforms)
bob/ip/binseg/configs/datasets/hrf544.py
+ 13
− 11
View file @ 833335da
......@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Resize((363))
,Pad((0,90,0,91))
,RandomRotation()
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
Resize((363)),
Pad((0, 90, 0, 91)),
RandomRotation(),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# 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
#### Config ####
transforms = Compose([
Resize((363))
,Pad((0,90,0,91))
,ToTensor()
])
transforms = Compose([Resize((363)), Pad((0, 90, 0, 91)), ToTensor()])
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# PyTorch dataset
dataset = BinSegDataset(bobdb, split='test', transform=transforms)
\ No newline at end of file
dataset = BinSegDataset(bobdb, split="test", transform=transforms)
bob/ip/binseg/configs/datasets/hrf608.py
+ 13
− 11
View file @ 833335da
......@@ -7,18 +7,20 @@ from bob.ip.binseg.data.binsegdataset import BinSegDataset
#### Config ####
transforms = Compose([
Pad((0,345,0,345))
,Resize(608)
,RandomRotation()
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
Pad((0, 345, 0, 345)),
Resize(608),
RandomRotation(),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# 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
#### Config ####
transforms = Compose([
Pad((0,584,0,584))
,Resize((960))
,RandomRotation()
,RandomHFlip()
,RandomVFlip()
,ColorJitter()
,ToTensor()
])
transforms = Compose(
[
Pad((0, 584, 0, 584)),
Resize((960)),
RandomRotation(),
RandomHFlip(),
RandomVFlip(),
ColorJitter(),
ToTensor(),
]
)
# bob.db.dataset init
bobdb = HRF(protocol = 'default')
bobdb = HRF(protocol="default")
# 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