-
Notifications
You must be signed in to change notification settings - Fork 0
/
cfg_training_data_creation.py
48 lines (42 loc) · 1.15 KB
/
cfg_training_data_creation.py
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": 1
},
{
"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
}