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_pc/__init__.py: 100%

1#!/usr/bin/env python 

2# coding=utf-8 

3 

4from torch.utils.data.dataset import ConcatDataset 

5 

6def _maker(protocol): 

7 

8 if protocol == "idiap": 

9 from ..nih_cxr14_re import idiap as nih_cxr14_re 

10 from ..padchest import no_tb_idiap as padchest_no_tb 

11 

12 nih_cxr14_re = nih_cxr14_re.dataset 

13 padchest_no_tb = padchest_no_tb.dataset 

14 

15 dataset = {} 

16 dataset['__train__'] = ConcatDataset([nih_cxr14_re["__train__"], padchest_no_tb["__train__"]]) 

17 dataset['train'] = ConcatDataset([nih_cxr14_re["train"], padchest_no_tb["train"]]) 

18 dataset['__valid__'] = ConcatDataset([nih_cxr14_re["__valid__"], padchest_no_tb["__valid__"]]) 

19 dataset['validation'] = ConcatDataset([nih_cxr14_re["validation"], padchest_no_tb["validation"]]) 

20 dataset['test'] = nih_cxr14_re["test"] 

21 

22 return dataset