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_RS import default as mc
10 from ..shenzhen_RS import default as ch
11 from ..indian_RS import default as indian
12 from ..padchest_RS import tb_idiap as pc
13
14 mc = mc.dataset
15 ch = ch.dataset
16 indian = indian.dataset
17 pc = pc.dataset
18
19 dataset = {}
20 dataset['__train__'] = ConcatDataset([mc["__train__"], ch["__train__"], indian["__train__"], pc["__train__"]])
21 dataset['train'] = ConcatDataset([mc["train"], ch["train"], indian["train"], pc["train"]])
22 dataset['__valid__'] = ConcatDataset([mc["__valid__"], ch["__valid__"], indian["__valid__"], pc["__valid__"]])
23 dataset['validation'] = ConcatDataset([mc["validation"], ch["validation"], indian["validation"], pc["validation"]])
24 dataset['test'] = ConcatDataset([mc["test"], ch["test"], indian["test"], pc["test"]])
25
26 return dataset