|
a |
|
b/libs/configs/config_acdc.py |
|
|
1 |
from easydict import EasyDict as edict |
|
|
2 |
import numpy as np |
|
|
3 |
|
|
|
4 |
__C = edict() |
|
|
5 |
cfg = __C |
|
|
6 |
|
|
|
7 |
|
|
|
8 |
# ============== general training config ===================== |
|
|
9 |
__C.TRAIN = edict() |
|
|
10 |
|
|
|
11 |
# __C.TRAIN.NET = "unet.U_Net" |
|
|
12 |
__C.TRAIN.NET = "unet_df.U_NetDF" |
|
|
13 |
|
|
|
14 |
__C.TRAIN.LR = 0.001 |
|
|
15 |
__C.TRAIN.LR_CLIP = 0.00001 |
|
|
16 |
__C.TRAIN.DECAY_STEP_LIST = [60, 100, 150, 180] |
|
|
17 |
__C.TRAIN.LR_DECAY = 0.5 |
|
|
18 |
|
|
|
19 |
__C.TRAIN.GRAD_NORM_CLIP = 1.0 |
|
|
20 |
|
|
|
21 |
__C.TRAIN.OPTIMIZER = 'adam' |
|
|
22 |
__C.TRAIN.WEIGHT_DECAY = 0 # "L2 regularization coeff [default: 0.0]" |
|
|
23 |
__C.TRAIN.MOMENTUM = 0.9 |
|
|
24 |
|
|
|
25 |
# =============== model config ======================== |
|
|
26 |
__C.MODEL = edict() |
|
|
27 |
|
|
|
28 |
__C.MODEL.SELFEATURE = True |
|
|
29 |
__C.MODEL.SHIFT_N = 1 |
|
|
30 |
__C.MODEL.AUXSEG = True |
|
|
31 |
|
|
|
32 |
# ================= dataset config ========================== |
|
|
33 |
__C.DATASET = edict() |
|
|
34 |
|
|
|
35 |
__C.DATASET.NAME = "acdc" |
|
|
36 |
__C.DATASET.MEAN = 63.19523533061758 |
|
|
37 |
__C.DATASET.STD = 70.74166957523165 |
|
|
38 |
|
|
|
39 |
__C.DATASET.NUM_CLASS = 4 |
|
|
40 |
|
|
|
41 |
__C.DATASET.DF_USED = True |
|
|
42 |
__C.DATASET.DF_NORM = True |
|
|
43 |
__C.DATASET.BOUNDARY = False |
|
|
44 |
|
|
|
45 |
__C.DATASET.TRAIN_LIST = "libs/datasets/jsonLists/acdcList/train.json" |
|
|
46 |
__C.DATASET.TEST_LIST = "libs/datasets/jsonLists/acdcList/test.json" |
|
|
47 |
|
|
|
48 |
__C.DATASET.TEST_PERSON_LIST = "libs/datasets/personList/AcdcTestPersonCarname.json" |