@@ 14-30 (lines=17) @@ | ||
11 | ||
12 | ||
13 | def get_loader(loader_name): |
|
14 | if loader_name in [ |
|
15 | "paired", |
|
16 | "unpaired", |
|
17 | "grouped", |
|
18 | ]: |
|
19 | dir_paths = [f"./data/test/nifti/{loader_name}/test"] |
|
20 | name = "fixed_images" if loader_name == "paired" else "images" |
|
21 | grouped = loader_name == "grouped" |
|
22 | elif loader_name == "multi_dirs_grouped": |
|
23 | dir_paths = [ |
|
24 | "./data/test/nifti/grouped/train", |
|
25 | "./data/test/nifti/grouped/test", |
|
26 | ] |
|
27 | name = "images" |
|
28 | grouped = True |
|
29 | else: |
|
30 | raise ValueError |
|
31 | loader = NiftiFileLoader(dir_paths=dir_paths, name=name, grouped=grouped) |
|
32 | return loader |
|
33 |
@@ 20-33 (lines=14) @@ | ||
17 | ||
18 | ||
19 | def get_loader(loader_name): |
|
20 | if loader_name in [ |
|
21 | "paired", |
|
22 | "unpaired", |
|
23 | "grouped", |
|
24 | ]: |
|
25 | dir_paths = [f"./data/test/h5/{loader_name}/test"] |
|
26 | name = "fixed_images" if loader_name == "paired" else "images" |
|
27 | grouped = loader_name == "grouped" |
|
28 | elif loader_name == "multi_dirs_grouped": |
|
29 | dir_paths = ["./data/test/h5/grouped/train", "./data/test/h5/grouped/test"] |
|
30 | name = "images" |
|
31 | grouped = True |
|
32 | else: |
|
33 | raise ValueError |
|
34 | loader = H5FileLoader(dir_paths=dir_paths, name=name, grouped=grouped) |
|
35 | return loader |
|
36 |