Coverage for src/deepdraw/configs/datasets/iostar/__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.iostar import dataset as raw 

8 from .. import make_dataset as mk 

9 

10 return mk(raw.subsets(protocol), []) 

11 

12 

13def _maker_square(protocol, size): 

14 from ....data.iostar import dataset as raw 

15 from ....data.transforms import Resize 

16 from .. import make_dataset as mk 

17 

18 return mk(raw.subsets(protocol), [Resize((size, size))])