forked from liliangqi/person_search
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
42 lines (36 loc) · 990 Bytes
/
config.yml
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
gamma: 0.1
disp_interval: 20
target_size: 600
max_size: 1000
pixel_means: [102.9801, 115.9465, 122.7717]
res50_fixed_blocks: 1
dense121_fixed_blocks: 1
rpn_channels: 512
anchor_scales: [8, 16, 32]
anchor_ratios: [0.5, 1, 2]
rpn_feat_stride: [16]
pooling_size: 7
train_rpn_pre_nms_top_n: 12000
train_rpn_post_nms_top_n: 2000
train_rpn_nms_thresh: 0.7
train_rpn_batchsize: 256
train_rpn_fg_frac: 0.5
train_rpn_pos_overlap: 0.7
train_rpn_neg_overlap: 0.3
train_rpn_clobber_positive: False
train_rpn_bbox_inside_weights: [1.0, 1.0, 1.0, 1.0]
train_rpn_pos_weight: -1.0
train_batch_size: 128
train_fg_frac: 0.25
train_fg_thresh: 0.5
train_bg_thresh_hi: 0.5
train_bg_thresh_lo: 0.0
train_bbox_normalize_targets_precomputed: True
train_bbox_normalize_means: [0.0, 0.0, 0.0, 0.0]
train_bbox_normalize_stds: [0.1, 0.1, 0.2, 0.2]
train_bbox_inside_weights: [1.0, 1.0, 1.0, 1.0]
test_rpn_pre_nms_top_n: 6000
test_rpn_post_nms_top_n: 300
test_rpn_nms_thresh: 0.7
test_bbox_reg: True
test_nms: 0.3