131 lines
2.5 KiB
YAML
131 lines
2.5 KiB
YAML
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
|