Upload conf_frcnn_layout.yaml
Browse files- conf_frcnn_layout.yaml +117 -0
conf_frcnn_layout.yaml
ADDED
|
@@ -0,0 +1,117 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
TAG: tp_casc_rcnn_X_32xd4_50_FPN_GN_2FC
|
| 2 |
+
BACKBONE:
|
| 3 |
+
BOTTLENECK: resnext_32xd4
|
| 4 |
+
FREEZE_AFFINE: false
|
| 5 |
+
FREEZE_AT: 0
|
| 6 |
+
NORM: GN
|
| 7 |
+
RESNET_NUM_BLOCKS:
|
| 8 |
+
- 3
|
| 9 |
+
- 4
|
| 10 |
+
- 6
|
| 11 |
+
- 3
|
| 12 |
+
TF_PAD_MODE: false
|
| 13 |
+
CASCADE:
|
| 14 |
+
BBOX_REG_WEIGHTS:
|
| 15 |
+
- - 10.0
|
| 16 |
+
- 10.0
|
| 17 |
+
- 5.0
|
| 18 |
+
- 5.0
|
| 19 |
+
- - 20.0
|
| 20 |
+
- 20.0
|
| 21 |
+
- 10.0
|
| 22 |
+
- 10.0
|
| 23 |
+
- - 30.0
|
| 24 |
+
- 30.0
|
| 25 |
+
- 15.0
|
| 26 |
+
- 15.0
|
| 27 |
+
IOUS:
|
| 28 |
+
- 0.5
|
| 29 |
+
- 0.6
|
| 30 |
+
- 0.7
|
| 31 |
+
DATA:
|
| 32 |
+
TRAIN_NUM_WORKERS: 12
|
| 33 |
+
FPN:
|
| 34 |
+
ANCHOR_STRIDES: !!python/tuple
|
| 35 |
+
- 4
|
| 36 |
+
- 8
|
| 37 |
+
- 16
|
| 38 |
+
- 32
|
| 39 |
+
- 64
|
| 40 |
+
CASCADE: true
|
| 41 |
+
FRCNN_CONV_HEAD_DIM: 256
|
| 42 |
+
FRCNN_FC_HEAD_DIM: 1024
|
| 43 |
+
FRCNN_HEAD_FUNC: fastrcnn_2fc_head
|
| 44 |
+
MRCNN_HEAD_FUNC: None
|
| 45 |
+
NORM: GN
|
| 46 |
+
NUM_CHANNEL: 256
|
| 47 |
+
PROPOSAL_MODE: Level
|
| 48 |
+
FRCNN:
|
| 49 |
+
BATCH_PER_IM: 512
|
| 50 |
+
BBOX_REG_WEIGHTS:
|
| 51 |
+
- 10.0
|
| 52 |
+
- 10.0
|
| 53 |
+
- 5.0
|
| 54 |
+
- 5.0
|
| 55 |
+
FG_RATIO: 0.25
|
| 56 |
+
FG_THRESH: 0.5
|
| 57 |
+
MODE_MASK: false
|
| 58 |
+
MRCNN:
|
| 59 |
+
ACCURATE_PASTE: true
|
| 60 |
+
HEAD_DIM: 256
|
| 61 |
+
PREPROC:
|
| 62 |
+
MAX_SIZE: 1333
|
| 63 |
+
PIXEL_MEAN:
|
| 64 |
+
- 238.234
|
| 65 |
+
- 238.14
|
| 66 |
+
- 238.145
|
| 67 |
+
PIXEL_STD:
|
| 68 |
+
- 7.961
|
| 69 |
+
- 7.876
|
| 70 |
+
- 7.81
|
| 71 |
+
SHORT_EDGE_SIZE: 800
|
| 72 |
+
TRAIN_SHORT_EDGE_SIZE:
|
| 73 |
+
- 800
|
| 74 |
+
- 1200
|
| 75 |
+
RPN:
|
| 76 |
+
ANCHOR_RATIOS: !!python/tuple
|
| 77 |
+
- 0.5
|
| 78 |
+
- 1.0
|
| 79 |
+
- 2.0
|
| 80 |
+
ANCHOR_SIZES: !!python/tuple
|
| 81 |
+
- 32
|
| 82 |
+
- 64
|
| 83 |
+
- 128
|
| 84 |
+
- 256
|
| 85 |
+
- 512
|
| 86 |
+
ANCHOR_STRIDE: 16
|
| 87 |
+
BATCH_PER_IM: 256
|
| 88 |
+
CROWD_OVERLAP_THRESH: 9.99
|
| 89 |
+
FG_RATIO: 0.5
|
| 90 |
+
HEAD_DIM: 1024
|
| 91 |
+
MIN_SIZE: 0
|
| 92 |
+
NEGATIVE_ANCHOR_THRESH: 0.3
|
| 93 |
+
POSITIVE_ANCHOR_THRESH: 0.7
|
| 94 |
+
PROPOSAL_NMS_THRESH: 0.7
|
| 95 |
+
PER_LEVEL_NMS_TOPK: 1000
|
| 96 |
+
TRAIN_PER_LEVEL_NMS_TOPK: 2000
|
| 97 |
+
TRAIN_PRE_NMS_TOPK: 12000
|
| 98 |
+
PRE_NMS_TOPK: 6000
|
| 99 |
+
TRAIN_POST_NMS_TOPK: 2000
|
| 100 |
+
POST_NMS_TOPK: 1000
|
| 101 |
+
OUTPUT:
|
| 102 |
+
FRCNN_NMS_THRESH: 0.001
|
| 103 |
+
RESULTS_PER_IM: 100
|
| 104 |
+
RESULT_SCORE_THRESH: 0.1
|
| 105 |
+
NMS_THRESH_CLASS_AGNOSTIC: 0.001
|
| 106 |
+
TRAINER: replicated
|
| 107 |
+
TRAIN:
|
| 108 |
+
LR_SCHEDULE: 4x
|
| 109 |
+
EVAL_PERIOD: 50
|
| 110 |
+
CHECKPOINT_PERIOD: 20
|
| 111 |
+
WEIGHT_DECAY: 1e-4
|
| 112 |
+
BASE_LR: 1e-2
|
| 113 |
+
WARMUP: 1000
|
| 114 |
+
WARMUP_INIT_LR: 1e-5
|
| 115 |
+
STEPS_PER_EPOCH: 500
|
| 116 |
+
STARTING_EPOCH: 1
|
| 117 |
+
LOG_DIR: train_log/maskrcnn
|