1234567891011121314151617181920212223242526272829303132333435363738 |
- architecture: SSD
- pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet34_pretrained.pdparams
- SSD:
- backbone: ResNet
- ssd_head: SSDHead
- post_process: BBoxPostProcess
- r34_backbone: True
- ResNet:
- # index 0 stands for res2
- depth: 34
- norm_type: bn
- freeze_norm: False
- freeze_at: -1
- return_idx: [2]
- num_stages: 3
- SSDHead:
- anchor_generator:
- steps: [8, 16, 32, 64, 100, 300]
- aspect_ratios: [[2.], [2., 3.], [2., 3.], [2., 3.], [2.], [2.]]
- min_sizes: [21.0, 45.0, 99.0, 153.0, 207.0, 261.0]
- max_sizes: [45.0, 99.0, 153.0, 207.0, 261.0, 315.0]
- offset: 0.5
- clip: True
- min_max_aspect_ratios_order: True
- use_extra_head: True
- BBoxPostProcess:
- decode:
- name: SSDBox
- nms:
- name: MultiClassNMS
- keep_top_k: 200
- score_threshold: 0.05
- nms_threshold: 0.5
- nms_top_k: 400
|