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-DRIVE + SSL for Vessel Segmentation
6* Configuration resolution: 544 x 544
8The dataset available in this file is composed of STARE, CHASE-DB1, IOSTAR
9vessel and HRF (with annotated samples) and DRIVE "default" training set
10without labels, for training, and DRIVE's "default" test set, for evaluation.
12For details on datasets, consult:
14* :py:mod:`bob.ip.binseg.data.stare`
15* :py:mod:`bob.ip.binseg.data.drive`
16* :py:mod:`bob.ip.binseg.data.chasedb1`
17* :py:mod:`bob.ip.binseg.data.iostar`
18* :py:mod:`bob.ip.binseg.data.hrf`
19"""
21from bob.ip.binseg.configs.datasets.drive.covd import dataset as _covd
22from bob.ip.binseg.configs.datasets.drive.default import dataset as _baseline
23from bob.ip.binseg.data.utils import SSLDataset
25# copy dictionary and replace only the augmented train dataset
26dataset = dict(**_covd)
27dataset["__train__"] = SSLDataset(_covd["__train__"], _baseline["__train__"])