Hot-keys on this page
r m x p toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1#!/usr/bin/env python
2# coding=utf-8
4"""COVD-CHASEDB1 for Vessel Segmentation
6* Configuration resolution (height x width): 960 x 960
8The dataset available in this file is composed of DRIVE, STARE, IOSTAR
9vessel and HRF (with annotated samples).
11For details on those datasets, consult:
13* See :py:mod:`bob.ip.binseg.data.drive`
14* See :py:mod:`bob.ip.binseg.data.stare`
15* See :py:mod:`bob.ip.binseg.data.iostar`
16* See :py:mod:`bob.ip.binseg.data.hrf`
17"""
19from torch.utils.data import ConcatDataset
21from bob.ip.binseg.configs.datasets import augment_subset as _augment
22from bob.ip.binseg.configs.datasets.chasedb1.first_annotator import (
23 dataset as _baseline,
24)
25from bob.ip.binseg.configs.datasets.chasedb1.first_annotator import (
26 second_annotator,
27)
28from bob.ip.binseg.configs.datasets.chasedb1.mtest import dataset as _mtest
30dataset = dict(**_baseline)
31dataset["__train__"] = ConcatDataset(
32 [
33 _augment(_mtest["drive (train)"], rotation_before=True),
34 _augment(_mtest["drive (test)"], rotation_before=True),
35 _augment(_mtest["stare (train)"], rotation_before=True),
36 _augment(_mtest["stare (test)"], rotation_before=True),
37 _augment(_mtest["hrf (train)"], rotation_before=False),
38 _augment(_mtest["hrf (test)"], rotation_before=False),
39 _augment(_mtest["iostar (train)"], rotation_before=False),
40 _augment(_mtest["iostar (test)"], rotation_before=False),
41 ]
42)
43del second_annotator["train"] # mismatch with used train set
44dataset["train"] = ConcatDataset(
45 [
46 _mtest["drive (train)"],
47 _mtest["drive (test)"],
48 _mtest["stare (train)"],
49 _mtest["stare (test)"],
50 _mtest["hrf (train)"],
51 _mtest["hrf (test)"],
52 _mtest["iostar (train)"],
53 _mtest["iostar (test)"],
54 ]
55)
56dataset["__valid__"] = dataset["train"]