Coverage for src/deepdraw/configs/datasets/drhagis/__init__.py: 100%

5 statements  

« prev     ^ index     » next       coverage.py v7.3.1, created at 2023-11-30 15:00 +0100

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.drhagis import dataset as raw 

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

9 from .. import make_dataset as mk 

10 

11 return mk( 

12 raw.subsets(protocol), 

13 [ShrinkIntoSquare(reference=2, threshold=0), Resize((1760, 1760))], 

14 )