69 lines
1.3 KiB
YAML
69 lines
1.3 KiB
YAML
architecture: FasterRCNN
|
|
pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/HRNet_W18_C_pretrained.pdparams
|
|
|
|
FasterRCNN:
|
|
backbone: HRNet
|
|
neck: HRFPN
|
|
rpn_head: RPNHead
|
|
bbox_head: BBoxHead
|
|
# post process
|
|
bbox_post_process: BBoxPostProcess
|
|
|
|
HRNet:
|
|
width: 18
|
|
freeze_at: 0
|
|
return_idx: [0, 1, 2, 3]
|
|
|
|
HRFPN:
|
|
out_channel: 256
|
|
share_conv: false
|
|
|
|
RPNHead:
|
|
anchor_generator:
|
|
aspect_ratios: [0.5, 1.0, 2.0]
|
|
anchor_sizes: [[32], [64], [128], [256], [512]]
|
|
strides: [4, 8, 16, 32, 64]
|
|
rpn_target_assign:
|
|
batch_size_per_im: 256
|
|
fg_fraction: 0.5
|
|
negative_overlap: 0.3
|
|
positive_overlap: 0.7
|
|
use_random: True
|
|
train_proposal:
|
|
min_size: 0.0
|
|
nms_thresh: 0.7
|
|
pre_nms_top_n: 2000
|
|
post_nms_top_n: 2000
|
|
topk_after_collect: True
|
|
test_proposal:
|
|
min_size: 0.0
|
|
nms_thresh: 0.7
|
|
pre_nms_top_n: 1000
|
|
post_nms_top_n: 1000
|
|
|
|
BBoxHead:
|
|
head: TwoFCHead
|
|
roi_extractor:
|
|
resolution: 7
|
|
sampling_ratio: 0
|
|
aligned: True
|
|
bbox_assigner: BBoxAssigner
|
|
|
|
BBoxAssigner:
|
|
batch_size_per_im: 512
|
|
bg_thresh: 0.5
|
|
fg_thresh: 0.5
|
|
fg_fraction: 0.25
|
|
use_random: True
|
|
|
|
TwoFCHead:
|
|
out_channel: 1024
|
|
|
|
BBoxPostProcess:
|
|
decode: RCNNBox
|
|
nms:
|
|
name: MultiClassNMS
|
|
keep_top_k: 100
|
|
score_threshold: 0.05
|
|
nms_threshold: 0.5
|