-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsap_R_50_C4_1x-city2foggy.yaml
46 lines (46 loc) · 1.14 KB
/
sap_R_50_C4_1x-city2foggy.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
_BASE_: "./Base-RCNN-C4.yaml"
DATASETS:
SOURCE_DOMAIN:
TRAIN: ("cityscapes_train",)
TARGET_DOMAIN:
TRAIN: ("foggy-cityscapes_train",)
TRAIN: ("cityscapes_train",)
TEST: ("foggy-cityscapes_val",)
MODEL:
META_ARCHITECTURE: "SAPRCNN"
BACKBONE:
FREEZE_AT: 1
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
KEYPOINT_ON: False
MASK_ON: False
DOMAIN_ADAPTATION_ON: True
ROI_HEADS:
NUM_CLASSES: 8
DA_HEAD:
IN_FEATURE: "res4"
IN_CHANNELS: 1024
NUM_ANCHOR_IN_IMG: 15
EMBEDDING_KERNEL_SIZE: 3
EMBEDDING_NORM: True
EMBEDDING_DROPOUT: True
FUNC_NAME: "cross_entropy"
POOL_TYPE: 'avg'
LOSS_WEIGHT: 1.0
WINDOW_STRIDES: [2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2]
WINDOW_SIZES: [3, 6, 9, 12, 15, 18, 21, 24, 27, 30, 33, 35, 37]
PROPOSAL_GENERATOR:
NAME: "SAPRPN"
INPUT:
MIN_SIZE_TRAIN: (800, 832, 864, 864, 896, 928, 960, 992, 1024)
MIN_SIZE_TRAIN_SAMPLING: "choice"
MIN_SIZE_TEST: 1024
MAX_SIZE_TRAIN: 2048
MAX_SIZE_TEST: 2048
SOLVER:
IMS_PER_BATCH: 1 # batch size
STEPS: (70000, 80000)
BASE_LR: 0.00001
MAX_ITER: 90000
CHECKPOINT_PERIOD: 5000
TEST:
EVAL_PERIOD: 1000