Coverage for src/deepdraw/configs/datasets/stare/mtest.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"""STARE cross-evaluation dataset with matched resolution. 

6 

7* Configuration resolution: 704 x 608 

8""" 

9 

10from ....data.transforms import CenterCrop, Pad, Resize 

11from .xtest import dataset as _xt 

12from .xtest import second_annotator # noqa 

13 

14dataset = { 

15 "train": _xt["train"], 

16 "test": _xt["test"], 

17 "drive (train)": _xt["drive (train)"].copy( 

18 [CenterCrop((470, 544)), Pad((10, 9, 10, 8)), Resize(608)] 

19 ), 

20 "drive (test)": _xt["drive (test)"].copy( 

21 [CenterCrop((470, 544)), Pad((10, 9, 10, 8)), Resize(608)] 

22 ), 

23 "chasedb1 (train)": _xt["chasedb1 (train)"].copy( 

24 [CenterCrop((829, 960)), Resize(608)] 

25 ), 

26 "chasedb1 (test)": _xt["chasedb1 (test)"].copy( 

27 [CenterCrop((829, 960)), Resize(608)] 

28 ), 

29 "hrf (train)": _xt["hrf (train)"].copy( 

30 [Pad((0, 345, 0, 345)), Resize(608)] 

31 ), 

32 "hrf (test)": _xt["hrf (test)"].copy([Pad((0, 345, 0, 345)), Resize(608)]), 

33 "iostar (train)": _xt["iostar (train)"].copy( 

34 [Pad((81, 0, 81, 0)), Resize(608)] 

35 ), 

36 "iostar (test)": _xt["iostar (test)"].copy( 

37 [Pad((81, 0, 81, 0)), Resize(608)] 

38 ), 

39}