Coverage for src/deepdraw/configs/datasets/iostar/vessel_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"""IOSTAR vessel cross-evaluation dataset with matched resolution. 

6 

7* Configuration resolution: 1024 x 1024 

8""" 

9 

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

11from .vessel_xtest import dataset as _xt 

12 

13dataset = { 

14 "train": _xt["train"], 

15 "test": _xt["test"], 

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

17 [CenterCrop((540, 540)), Resize(1024)] 

18 ), 

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

20 [CenterCrop((540, 540)), Resize(1024)] 

21 ), 

22 "stare (train)": _xt["stare (train)"].copy( 

23 [Pad((0, 32, 0, 32)), Resize(1024), CenterCrop(1024)] 

24 ), 

25 "stare (test)": _xt["stare (test)"].copy( 

26 [Pad((0, 32, 0, 32)), Resize(1024), CenterCrop(1024)] 

27 ), 

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

29 [Crop(0, 18, 960, 960), Resize(1024)] 

30 ), 

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

32 [Crop(0, 18, 960, 960), Resize(1024)] 

33 ), 

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

35 [Pad((0, 584, 0, 584)), Resize(1024)] 

36 ), 

37 "hrf (test)": _xt["hrf (test)"].copy([Pad((0, 584, 0, 584)), Resize(1024)]), 

38}