CKPT_PERIOD: 1
DEVICE: cuda:0
DISP_PERIOD: 10
EVAL_FEATURE: concat
EVAL_GALLERY_SIZE: 100
EVAL_PERIOD: 1
EVAL_USE_CACHE: false
EVAL_USE_CBGM: false
EVAL_USE_GT: false
GRID:
  MODE: 1
  OFFSET: 0
  PROB: 0.5
  RATIO: 0.5
  ROTATE: 1
INPUT:
  BATCH_SIZE_TEST: 1
  BATCH_SIZE_TRAIN: 1
  DATASET: CUHK-SYSU
  DATA_ROOT: E:/DeepLearning/PersonSearch/COAT/datasets/CUHK-SYSU
  IMAGE_CUTOUT: false
  IMAGE_ERASE: false
  IMAGE_GRID: false
  IMAGE_MIXUP: false
  MAX_SIZE: 1500
  MIN_SIZE: 900
  NUM_WORKERS_TEST: 1
  NUM_WORKERS_TRAIN: 5
MODEL:
  EMBEDDING_DIM: 256
  FEATURE_AUG_TYPE: exchange_token
  FEATURE_MASK_SIZE: 4
  LOSS:
    CQ_SIZE: 5000
    LUT_SIZE: 5532
    OIM_MOMENTUM: 0.5
    OIM_SCALAR: 30.0
    USE_SOFTMAX: true
  MASK_MODE: random_direction
  MASK_PERCENT: 0.1
  MASK_SHAPE: stripe
  MASK_SIZE: 1
  ROI_HEAD:
    BATCH_SIZE_TRAIN: 128
    BN_NECK: true
    DETECTIONS_PER_IMAGE_TEST: 300
    NEG_THRESH_TRAIN: 0.5
    NEG_THRESH_TRAIN_2ND: 0.6
    NEG_THRESH_TRAIN_3RD: 0.7
    NMS_THRESH_TEST: 0.4
    NMS_THRESH_TEST_1ST: 0.4
    NMS_THRESH_TEST_2ND: 0.4
    NMS_THRESH_TEST_3RD: 0.5
    POS_FRAC_TRAIN: 0.25
    POS_THRESH_TRAIN: 0.5
    POS_THRESH_TRAIN_2ND: 0.6
    POS_THRESH_TRAIN_3RD: 0.7
    SCORE_THRESH_TEST: 0.5
    USE_DIFF_THRESH: true
  RPN:
    BATCH_SIZE_TRAIN: 256
    NEG_THRESH_TRAIN: 0.3
    NMS_THRESH: 0.7
    POST_NMS_TOPN_TEST: 300
    POST_NMS_TOPN_TRAIN: 2000
    POS_FRAC_TRAIN: 0.5
    POS_THRESH_TRAIN: 0.7
    PRE_NMS_TOPN_TEST: 6000
    PRE_NMS_TOPN_TRAIN: 12000
  TRANSFORMER:
    DIM_MODEL: 512
    DROPOUT: 0.0
    ENCODER_LAYERS: 1
    KERNEL_SIZE_1ST:
    - &id001
      - 1
      - 1
    - &id002
      - 3
      - 3
    KERNEL_SIZE_2ND:
    - *id001
    - *id002
    KERNEL_SIZE_3RD:
    - *id001
    - *id002
    NAMES_1ST:
    - scale1
    - scale2
    NAMES_2ND:
    - scale1
    - scale2
    NAMES_3RD:
    - scale1
    - scale2
    N_HEAD: 8
    USE_DIFF_SCALE: true
    USE_GLOBAL_SHORTCUT: true
    USE_LOCAL_SHORTCUT: true
    USE_MASK_1ST: false
    USE_MASK_2ND: true
    USE_MASK_3RD: true
    USE_OUTPUT_LAYER: false
    USE_PATCH2VEC: true
  USE_FEATURE_MASK: true
OUTPUT_DIR: ./output
SEED: 1
SOLVER:
  BASE_LR: 0.003
  CLIP_GRADIENTS: 10.0
  GAMMA: 0.1
  LR_DECAY_MILESTONES:
  - 10
  - 14
  LW_RCNN_CLS_1ST: 1
  LW_RCNN_CLS_2ND: 1
  LW_RCNN_CLS_3RD: 1
  LW_RCNN_REG_1ST: 10
  LW_RCNN_REG_2ND: 10
  LW_RCNN_REG_3RD: 10
  LW_RCNN_REID_2ND: 0.5
  LW_RCNN_REID_3RD: 0.5
  LW_RCNN_SOFTMAX_2ND: 0.5
  LW_RCNN_SOFTMAX_3RD: 0.5
  LW_RPN_CLS: 1
  LW_RPN_REG: 1
  MAX_EPOCHS: 13
  SGD_MOMENTUM: 0.9
  WEIGHT_DECAY: 0.0005
TF_BOARD: true
