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/mc_ch_in_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 == "default": 

9 from ..montgomery import default as mc 

10 from ..shenzhen import default as ch 

11 from ..indian import default as indian 

12 from ..padchest import tb_idiap as pc 

13 elif protocol == "rgb": 

14 from ..montgomery import rgb as mc 

15 from ..shenzhen import rgb as ch 

16 from ..indian import rgb as indian 

17 from ..padchest import tb_idiap_rgb as pc 

18 

19 mc = mc.dataset 

20 ch = ch.dataset 

21 indian = indian.dataset 

22 pc = pc.dataset 

23 

24 dataset = {} 

25 dataset['__train__'] = ConcatDataset([mc["__train__"], ch["__train__"], indian["__train__"], pc["__train__"]]) 

26 dataset['train'] = ConcatDataset([mc["train"], ch["train"], indian["train"], pc["train"]]) 

27 dataset['__valid__'] = ConcatDataset([mc["__valid__"], ch["__valid__"], indian["__valid__"], pc["__valid__"]]) 

28 dataset['test'] = ConcatDataset([mc["test"], ch["test"], indian["test"], pc["test"]]) 

29 

30 return dataset