1#!/usr/bin/env python 

2# coding=utf-8 

3 

4"""COVD-DRIVE for Vessel Segmentation 

5 

6* Configuration resolution: 544 x 544 

7 

8The dataset available in this file is composed of STARE, CHASE-DB1, IOSTAR 

9vessel and HRF (with annotated samples). 

10 

11For details on those datasets, consult: 

12 

13* See :py:mod:`bob.ip.binseg.data.stare` 

14* See :py:mod:`bob.ip.binseg.data.chasedb1` 

15* See :py:mod:`bob.ip.binseg.data.iostar` 

16* See :py:mod:`bob.ip.binseg.data.hrf` 

17""" 

18 

19from torch.utils.data import ConcatDataset 

20 

21from bob.ip.binseg.configs.datasets import augment_subset as _augment 

22from bob.ip.binseg.configs.datasets.drive.default import dataset as _baseline 

23from bob.ip.binseg.configs.datasets.drive.mtest import dataset as _mtest 

24 

25dataset = dict(**_baseline) 

26dataset["__train__"] = ConcatDataset( 

27 [ 

28 _augment(_mtest["stare (train)"], rotation_before=True), 

29 _augment(_mtest["stare (test)"], rotation_before=True), 

30 _augment(_mtest["chasedb1 (train)"], rotation_before=False), 

31 _augment(_mtest["chasedb1 (test)"], rotation_before=False), 

32 _augment(_mtest["hrf (train)"], rotation_before=False), 

33 _augment(_mtest["hrf (test)"], rotation_before=False), 

34 _augment(_mtest["iostar (train)"], rotation_before=False), 

35 _augment(_mtest["iostar (test)"], rotation_before=False), 

36 ] 

37) 

38dataset["train"] = ConcatDataset( 

39 [ 

40 _mtest["stare (train)"], 

41 _mtest["stare (test)"], 

42 _mtest["chasedb1 (train)"], 

43 _mtest["chasedb1 (test)"], 

44 _mtest["hrf (train)"], 

45 _mtest["hrf (test)"], 

46 _mtest["iostar (train)"], 

47 _mtest["iostar (test)"], 

48 ] 

49) 

50dataset["__valid__"] = dataset["train"]