Coverage for /scratch/builds/bob/bob.med.tb/miniconda/conda-bld/bob.med.tb_1637571489937/_test_env_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placehold_placeho/lib/python3.8/site-packages/bob/med/tb/configs/datasets/nih_cxr14_re/__init__.py: 100%

1#!/usr/bin/env python 

2# coding=utf-8 

3 

4def _maker(protocol, size=512): 

5 

6 from ....data.transforms import RemoveBlackBorders 

7 import torchvision.transforms as transforms 

8 from ....data.nih_cxr14_re import dataset as raw 

9 from .. import make_dataset as mk 

10 

11 # ImageNet normalization 

12 normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], 

13 std=[0.229, 0.224, 0.225]) 

14 

15 return mk( 

16 [raw.subsets(protocol)], 

17 [transforms.Resize((size, size))], 

18 [transforms.RandomHorizontalFlip()], 

19 [transforms.ToTensor(), normalize] 

20 )