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 elif protocol == "fold_0":
13 from ..montgomery_RS import fold_0 as mc
14 from ..shenzhen_RS import fold_0 as ch
15 from ..indian_RS import fold_0 as indian
16 elif protocol == "fold_1":
17 from ..montgomery_RS import fold_1 as mc
18 from ..shenzhen_RS import fold_1 as ch
19 from ..indian_RS import fold_1 as indian
20 elif protocol == "fold_2":
21 from ..montgomery_RS import fold_2 as mc
22 from ..shenzhen_RS import fold_2 as ch
23 from ..indian_RS import fold_2 as indian
24 elif protocol == "fold_3":
25 from ..montgomery_RS import fold_3 as mc
26 from ..shenzhen_RS import fold_3 as ch
27 from ..indian_RS import fold_3 as indian
28 elif protocol == "fold_4":
29 from ..montgomery_RS import fold_4 as mc
30 from ..shenzhen_RS import fold_4 as ch
31 from ..indian_RS import fold_4 as indian
32 elif protocol == "fold_5":
33 from ..montgomery_RS import fold_5 as mc
34 from ..shenzhen_RS import fold_5 as ch
35 from ..indian_RS import fold_5 as indian
36 elif protocol == "fold_6":
37 from ..montgomery_RS import fold_6 as mc
38 from ..shenzhen_RS import fold_6 as ch
39 from ..indian_RS import fold_6 as indian
40 elif protocol == "fold_7":
41 from ..montgomery_RS import fold_7 as mc
42 from ..shenzhen_RS import fold_7 as ch
43 from ..indian_RS import fold_7 as indian
44 elif protocol == "fold_8":
45 from ..montgomery_RS import fold_8 as mc
46 from ..shenzhen_RS import fold_8 as ch
47 from ..indian_RS import fold_8 as indian
48 elif protocol == "fold_9":
49 from ..montgomery_RS import fold_9 as mc
50 from ..shenzhen_RS import fold_9 as ch
51 from ..indian_RS import fold_9 as indian
52
53 mc = mc.dataset
54 ch = ch.dataset
55 indian = indian.dataset
56
57 dataset = {}
58 dataset['__train__'] = ConcatDataset([mc["__train__"], ch["__train__"], indian["__train__"]])
59 dataset['train'] = ConcatDataset([mc["train"], ch["train"], indian["train"]])
60 dataset['__valid__'] = ConcatDataset([mc["__valid__"], ch["__valid__"], indian["__valid__"]])
61 dataset['validation'] = ConcatDataset([mc["validation"], ch["validation"], indian["validation"]])
62 dataset['test'] = ConcatDataset([mc["test"], ch["test"], indian["test"]])
63
64 return dataset