Coverage for src/deepdraw/configs/datasets/hrf/__init__.py: 68%

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

2# 

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

4 

5 

6def _maker_1168(protocol): 

7 from ....data.hrf import dataset as raw 

8 from ....data.transforms import Crop, Resize 

9 from .. import make_dataset as mk 

10 

11 return mk(raw.subsets(protocol), [Crop(0, 108, 2336, 3296), Resize(1168)]) 

12 

13 

14def _maker(protocol): 

15 from ....data.hrf import dataset as raw 

16 from ....data.transforms import Crop 

17 from .. import make_dataset as mk 

18 

19 return mk(raw.subsets(protocol), [Crop(0, 108, 2336, 3296)]) 

20 

21 

22def _maker_square_768(protocol): 

23 from ....data.hrf import dataset as raw 

24 from ....data.transforms import Pad, Resize 

25 from .. import make_dataset as mk 

26 

27 return mk(raw.subsets(protocol), [Pad((0, 584)), Resize((768, 768))]) 

28 

29 

30def _maker_square_1024(protocol): 

31 from ....data.hrf import dataset as raw 

32 from ....data.transforms import Pad, Resize 

33 from .. import make_dataset as mk 

34 

35 return mk(raw.subsets(protocol), [Pad((0, 584)), Resize((1024, 1024))]) 

36 

37 

38def _maker_square(protocol, size): 

39 from ....data.hrf import dataset as raw 

40 from ....data.transforms import Pad, Resize 

41 from .. import make_dataset as mk 

42 

43 return mk(raw.subsets(protocol), [Pad((0, 584)), Resize((size, size))])