-
Notifications
You must be signed in to change notification settings - Fork 27
/
3d_instance_segmentation.yaml
53 lines (46 loc) · 1.1 KB
/
3d_instance_segmentation.yaml
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
48
49
50
51
52
53
# BiaPy version: 3.4.3
SYSTEM:
NUM_CPUS: -1
NUM_WORKERS: 5
PROBLEM:
TYPE: INSTANCE_SEG
NDIM: 3D
INSTANCE_SEG:
DATA_CHANNELS: BC
DATA_MW_TH_TYPE: "auto"
DATA:
PATCH_SIZE: (20, 128, 128, 1)
TRAIN:
PATH: /path/to/data
GT_PATH: /path/to/data
IN_MEMORY: True
VAL:
SPLIT_TRAIN: 0.1
TEST:
PATH: /path/to/data
GT_PATH: /path/to/data
IN_MEMORY: True
LOAD_GT: True
PADDING: (4,16,16)
AUGMENTOR:
ENABLE: True
RANDOM_ROT: True
VFLIP: True
HFLIP: True
ZFLIP: True
MODEL:
ARCHITECTURE: unet
FEATURE_MAPS: [16, 32, 64, 128, 256]
Z_DOWN: [1,1,1,1]
LOAD_CHECKPOINT: False
TRAIN:
ENABLE: True
OPTIMIZER: ADAMW
LR: 1.E-4
BATCH_SIZE: 6
EPOCHS: 360
PATIENCE: 50
TEST:
ENABLE: True
AUGMENTATION: True
FULL_IMG: False