1#!/usr/bin/env python 

2# coding=utf-8 

3 

4"""COVD-CHASE-DB1 + SSL for Vessel Segmentation 

5 

6* Configuration resolution: 960 x 960 

7 

8The dataset available in this file is composed of DRIVE, STARE, IOSTAR vessel 

9and HRF (with annotated samples) and CHASE-DB1's "first-annotator" training set 

10without labels, for training, and CHASE-DB1's "first-annotator" test set, for 

11evaluation. 

12 

13For details on datasets, consult: 

14 

15* :py:mod:`bob.ip.binseg.data.stare` 

16* :py:mod:`bob.ip.binseg.data.drive` 

17* :py:mod:`bob.ip.binseg.data.chasedb1` 

18* :py:mod:`bob.ip.binseg.data.iostar` 

19* :py:mod:`bob.ip.binseg.data.hrf` 

20""" 

21 

22from bob.ip.binseg.configs.datasets.chasedb1.covd import dataset as _covd 

23from bob.ip.binseg.configs.datasets.chasedb1.first_annotator import ( 

24 dataset as _baseline, 

25) 

26from bob.ip.binseg.data.utils import SSLDataset 

27 

28# copy dictionary and replace only the augmented train dataset 

29dataset = dict(**_covd) 

30dataset["__train__"] = SSLDataset(_covd["__train__"], _baseline["__train__"])