-
Notifications
You must be signed in to change notification settings - Fork 21
/
da_ga_ca_cityscapes_VGG_16_FPN_4x.yaml
95 lines (95 loc) · 2.3 KB
/
da_ga_ca_cityscapes_VGG_16_FPN_4x.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
MODEL:
META_ARCHITECTURE: "GeneralizedRCNN"
# WEIGHT: 'https://s3.ap-northeast-2.amazonaws.com/open-mmlab/pretrain/third_party/vgg16_caffe-292e1171.pth' # Initialed by pretrained weight
#
WEIGHT: "./model_GA.pth" # Initialed by GA pretrained weight (Optimizer and scheduler should not be loaded)
#
RPN_ONLY: True
FCOS_ON: True
DA_ON: True
BACKBONE:
CONV_BODY: "VGG-16-FPN-RETINANET"
RETINANET:
USE_C5: False # FCOS uses P5 instead of C5
FCOS:
NUM_CLASSES: 9
INFERENCE_TH: 0.05 # pre_nms_thresh (default=0.05)
PRE_NMS_TOP_N: 1000 # pre_nms_top_n (default=1000)
NMS_TH: 0.6 # nms_thresh (default=0.6)
ADV:
#
USE_DIS_GLOBAL: True
USE_DIS_CENTER_AWARE: True
CENTER_AWARE_WEIGHT: 20
CENTER_AWARE_TYPE: "ca_feature" # or "ca_loss"
#
GA_DIS_LAMBDA: 0.01
CA_DIS_LAMBDA: 0.1
GRL_APPLIED_DOMAIN: 'both'
#
#
USE_DIS_P7: True
USE_DIS_P6: True
USE_DIS_P5: True
USE_DIS_P4: True
USE_DIS_P3: True
#
DIS_P7_NUM_CONVS: 4
DIS_P6_NUM_CONVS: 4
DIS_P5_NUM_CONVS: 4
DIS_P4_NUM_CONVS: 4
DIS_P3_NUM_CONVS: 4
#
CA_DIS_P7_NUM_CONVS: 4
CA_DIS_P6_NUM_CONVS: 4
CA_DIS_P5_NUM_CONVS: 4
CA_DIS_P4_NUM_CONVS: 4
CA_DIS_P3_NUM_CONVS: 4
#
GRL_WEIGHT_P7: 0.02
GRL_WEIGHT_P6: 0.02
GRL_WEIGHT_P5: 0.02
GRL_WEIGHT_P4: 0.02
GRL_WEIGHT_P3: 0.02
#
CA_GRL_WEIGHT_P7: 0.02
CA_GRL_WEIGHT_P6: 0.02
CA_GRL_WEIGHT_P5: 0.02
CA_GRL_WEIGHT_P4: 0.02
CA_GRL_WEIGHT_P3: 0.02
TEST:
DETECTIONS_PER_IMG: 100 # fpn_post_nms_top_n (default=100)
DATASETS:
TRAIN_SOURCE: ("cityscapes_train_cocostyle", )
TRAIN_TARGET: ("cityscapes_foggy_train_cocostyle", )
TEST: ("cityscapes_foggy_val_cocostyle", )
INPUT:
MIN_SIZE_RANGE_TRAIN: (640, 800)
MAX_SIZE_TRAIN: 1333
MIN_SIZE_TEST: 800
MAX_SIZE_TEST: 1333
DATALOADER:
SIZE_DIVISIBILITY: 32
SOLVER:
WEIGHT_DECAY: 0.0001
MAX_ITER: 20000
IMS_PER_BATCH: 16
CHECKPOINT_PERIOD: 500
#
BACKBONE:
BASE_LR: 0.005
STEPS: (120000, 160000)
WARMUP_ITERS: 500
WARMUP_METHOD: "constant"
#
FCOS:
BASE_LR: 0.005
STEPS: (120000, 160000)
WARMUP_ITERS: 500
WARMUP_METHOD: "constant"
#
DIS:
BASE_LR: 0.005
STEPS: (120000, 160000)
WARMUP_ITERS: 500
WARMUP_METHOD: "constant"