File size: 1,929 Bytes
8c34673
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
TAG: tp_casc_rcnn_X_32xd4_50_FPN_GN_2FC
BACKBONE:
  BOTTLENECK: resnext_32xd4
  FREEZE_AFFINE: false
  FREEZE_AT: 0
  NORM: GN
  RESNET_NUM_BLOCKS:
    - 3
    - 4
    - 6
    - 3
  TF_PAD_MODE: false
CASCADE:
  BBOX_REG_WEIGHTS:
    - - 10.0
      - 10.0
      - 5.0
      - 5.0
    - - 20.0
      - 20.0
      - 10.0
      - 10.0
    - - 30.0
      - 30.0
      - 15.0
      - 15.0
  IOUS:
    - 0.5
    - 0.6
    - 0.7
DATA:
  TRAIN_NUM_WORKERS: 12
FPN:
  ANCHOR_STRIDES: !!python/tuple
    - 4
    - 8
    - 16
    - 32
    - 64
  CASCADE: true
  FRCNN_CONV_HEAD_DIM: 256
  FRCNN_FC_HEAD_DIM: 1024
  FRCNN_HEAD_FUNC: fastrcnn_2fc_head
  NORM: GN
  NUM_CHANNEL: 256
  PROPOSAL_MODE: Level
FRCNN:
  BATCH_PER_IM: 512
  BBOX_REG_WEIGHTS:
    - 10.0
    - 10.0
    - 5.0
    - 5.0
  FG_RATIO: 0.25
  FG_THRESH: 0.5
MODE_MASK: false
MRCNN:
  ACCURATE_PASTE: true
  HEAD_DIM: 256
PREPROC:
  MAX_SIZE: 1333
  PIXEL_MEAN:
    - 238.33
    - 238.24
    - 238.23
  PIXEL_STD:
    - 8.75
    - 8.569
    - 9.14
  SHORT_EDGE_SIZE: 600
  TRAIN_SHORT_EDGE_SIZE:
    - 400
    - 600
RPN:
  ANCHOR_RATIOS: !!python/tuple
    - 0.25
    - 0.5
    - 1.0
  ANCHOR_SIZES: !!python/tuple
    - 8
    - 16
    - 32
    - 64
    - 128
  ANCHOR_STRIDE: 8
  BATCH_PER_IM: 512
  CROWD_OVERLAP_THRESH: 9.99
  FG_RATIO: 0.5
  HEAD_DIM: 1024
  MIN_SIZE: 0
  NEGATIVE_ANCHOR_THRESH: 0.3
  POSITIVE_ANCHOR_THRESH: 0.7
  PROPOSAL_NMS_THRESH: 0.7
  PER_LEVEL_NMS_TOPK: 1000
  TRAIN_PER_LEVEL_NMS_TOPK: 2000
  TRAIN_PRE_NMS_TOPK: 12000
  PRE_NMS_TOPK: 6000
  TRAIN_POST_NMS_TOPK: 2000
  POST_NMS_TOPK: 1000
OUTPUT:
  FRCNN_NMS_THRESH: 0.01
  RESULTS_PER_IM: 800
  RESULT_SCORE_THRESH: 0.4
  NMS_THRESH_CLASS_AGNOSTIC: 0.001
TRAINER: replicated
TRAIN:
  LR_SCHEDULE: 1x
  EVAL_PERIOD: 50
  CHECKPOINT_PERIOD: 20
  WEIGHT_DECAY: 1e-4
  BASE_LR: 1e-2
  WARMUP: 1000
  WARMUP_INIT_LR: 1e-5
  STEPS_PER_EPOCH: 500
  STARTING_EPOCH: 1
  LOG_DIR: train_log/maskrcnn