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 | |