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

6 

7* Configuration resolution: 1168 x 1648 

8""" 

9 

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

11from .xtest import dataset as _xt 

12 

13dataset = { 

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

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

16 "train (full resolution)": _xt["train (full resolution)"], 

17 "test (full resolution)": _xt["test (full resolution)"], 

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

19 [Crop(75, 10, 416, 544), Pad((21, 0, 22, 0)), Resize(1168)] 

20 ), 

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

22 [Crop(75, 10, 416, 544), Pad((21, 0, 22, 0)), Resize(1168)] 

23 ), 

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

25 [Crop(50, 0, 500, 705), Resize(1168), Pad((1, 0, 1, 0))] 

26 ), 

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

28 [Crop(50, 0, 500, 705), Resize(1168), Pad((1, 0, 1, 0))] 

29 ), 

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

31 [Crop(140, 18, 680, 960), Resize(1168)] 

32 ), 

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

34 [Crop(140, 18, 680, 960), Resize(1168)] 

35 ), 

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

37 [Crop(144, 0, 768, 1024), Pad((30, 0, 30, 0)), Resize(1168)] 

38 ), 

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

40 [Crop(144, 0, 768, 1024), Pad((30, 0, 30, 0)), Resize(1168)] 

41 ), 

42}