Coverage for src/deepdraw/configs/datasets/stare/covd.py: 100%

1# SPDX-FileCopyrightText: Copyright © 2023 Idiap Research Institute <contact@idiap.ch> 

2# 

3# SPDX-License-Identifier: GPL-3.0-or-later 

4 

5"""COVD-STARE for Vessel Segmentation. 

6 

7* Configuration resolution: 704 x 608 

8 

9The dataset available in this file is composed of DRIVE, CHASE-DB1, IOSTAR 

10vessel and HRF (with annotated samples). 

11 

12For details on those datasets, consult: 

13 

14* See :py:mod:`deepdraw.data.drive` 

15* See :py:mod:`deepdraw.data.chasedb1` 

16* See :py:mod:`deepdraw.data.iostar` 

17* See :py:mod:`deepdraw.data.hrf` 

18""" 

19 

20from torch.utils.data import ConcatDataset 

21 

22from .. import augment_subset as _augment 

23from .ah import dataset as _baseline 

24from .ah import second_annotator 

25from .mtest import dataset as _mtest 

26 

27dataset = dict(**_baseline) 

28dataset["__train__"] = ConcatDataset( 

29 [ 

30 _augment(_mtest["drive (train)"], rotation_before=True), 

31 _augment(_mtest["drive (test)"], rotation_before=True), 

32 _augment(_mtest["chasedb1 (train)"], rotation_before=True), 

33 _augment(_mtest["chasedb1 (test)"], rotation_before=True), 

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

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

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

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

38 ] 

39) 

40del second_annotator["train"] # mismatch with used train set 

41dataset["train"] = ConcatDataset( 

42 [ 

43 _mtest["drive (train)"], 

44 _mtest["drive (test)"], 

45 _mtest["chasedb1 (train)"], 

46 _mtest["chasedb1 (test)"], 

47 _mtest["hrf (train)"], 

48 _mtest["hrf (test)"], 

49 _mtest["iostar (train)"], 

50 _mtest["iostar (test)"], 

51 ] 

52) 

53dataset["__valid__"] = dataset["train"]