deepdraw.configs.datasets.drive.covd#
COVD-DRIVE for Vessel Segmentation.
Configuration resolution: 544 x 544
The dataset available in this file is composed of STARE, CHASE-DB1, 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 .default import dataset as _baseline
from .mtest import dataset as _mtest
dataset = dict(**_baseline)
dataset["__train__"] = ConcatDataset(
[
_augment(_mtest["stare (train)"], rotation_before=True),
_augment(_mtest["stare (test)"], rotation_before=True),
_augment(_mtest["chasedb1 (train)"], rotation_before=False),
_augment(_mtest["chasedb1 (test)"], rotation_before=False),
_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),
]
)
dataset["train"] = ConcatDataset(
[
_mtest["stare (train)"],
_mtest["stare (test)"],
_mtest["chasedb1 (train)"],
_mtest["chasedb1 (test)"],
_mtest["hrf (train)"],
_mtest["hrf (test)"],
_mtest["iostar (train)"],
_mtest["iostar (test)"],
]
)
dataset["__valid__"] = dataset["train"]