deepdraw.configs.datasets.chasedb1.covd#
COVD-CHASEDB1 for Vessel Segmentation.
Configuration resolution (height x width): 960 x 960
The dataset available in this file is composed of DRIVE, STARE, IOSTAR vessel and HRF (with annotated samples).
For details on those datasets, consult:
from torch.utils.data import ConcatDataset
from .. import augment_subset as _augment
from .first_annotator import dataset as _baseline
from .first_annotator import second_annotator
from .mtest import dataset as _mtest
dataset = dict(**_baseline)
dataset["__train__"] = ConcatDataset(
[
_augment(_mtest["drive (train)"], rotation_before=True),
_augment(_mtest["drive (test)"], rotation_before=True),
_augment(_mtest["stare (train)"], rotation_before=True),
_augment(_mtest["stare (test)"], rotation_before=True),
_augment(_mtest["hrf (train)"], rotation_before=False),
_augment(_mtest["hrf (test)"], rotation_before=False),
_augment(_mtest["iostar (train)"], rotation_before=False),
_augment(_mtest["iostar (test)"], rotation_before=False),
]
)
del second_annotator["train"] # mismatch with used train set
dataset["train"] = ConcatDataset(
[
_mtest["drive (train)"],
_mtest["drive (test)"],
_mtest["stare (train)"],
_mtest["stare (test)"],
_mtest["hrf (train)"],
_mtest["hrf (test)"],
_mtest["iostar (train)"],
_mtest["iostar (test)"],
]
)
dataset["__valid__"] = dataset["train"]