Coverage for src/deepdraw/configs/datasets/drishtigs1/__init__.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 

6def _maker(protocol): 

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

8 from ....data.transforms import CenterCrop as ccrop 

9 from .. import make_dataset as mk 

10 

11 return mk(raw.subsets(protocol), [ccrop((1760, 2048))]) 

12 

13 

14def _maker_square(protocol, size): 

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

16 from ....data.transforms import CenterCrop as ccrop 

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

18 from .. import make_dataset as mk 

19 

20 return mk( 

21 raw.subsets(protocol), 

22 [ccrop((1760, 2048)), Pad((0, 144)), Resize((size, size))], 

23 )