[26e314]: / cfg_training_data_creation.py

Download this file

49 lines (42 with data), 1.2 kB

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
__author__ = 'adeb'
data_path = "./datasets/paper_ultimate_conv/"
general = {
"source": "miccai",
"source_kwargs": {"mode": "folder", "path": "./datasets/miccai/1/"},
"file_path": data_path + "train.h5",
"perm": True,
"n_data": 320000
}
pick_vx = {
"where": "anywhere", # anywhere, plane
"plane": 100,
"axis": 1,
"how": "random" # all, random, balanced
}
pick_features = [
# types: 2Dortho, 3D, centroid
{
"how": "2Dortho", # 2Dortho, 2DorthoRotated, 3D, three_patches, local_global, grid_patches
"patch_width": 29,
"axis": [0, 1, 2],
"scale": 3
},
{
"how": "2Dortho", # 2Dortho, 2DorthoRotated, 3D, three_patches, local_global, grid_patches
"patch_width": 29,
"axis": [0, 1, 2],
"scale": 3
},
{
"how": "3D", # 2Dortho, 2DorthoRotated, 3D, three_patches, local_global, grid_patches
"patch_width": 13,
"scale": 1
},
{
"how": "centroid", # 2Dortho, 2DorthoRotated, 3D, three_patches, local_global, grid_patches
"n_features": 134
}
]
pick_tg = {
"how": "center" # center, proportion
}