centernet_r50.yml 614 B

12345678910111213141516171819202122232425262728293031323334
  1. architecture: CenterNet
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_vd_ssld_pretrained.pdparams
  3. norm_type: sync_bn
  4. use_ema: true
  5. ema_decay: 0.9998
  6. CenterNet:
  7. backbone: ResNet
  8. neck: CenterNetDLAFPN
  9. head: CenterNetHead
  10. post_process: CenterNetPostProcess
  11. ResNet:
  12. depth: 50
  13. variant: d
  14. return_idx: [0, 1, 2, 3]
  15. freeze_at: -1
  16. norm_decay: 0.
  17. dcn_v2_stages: [3]
  18. CenterNetDLAFPN:
  19. first_level: 0
  20. last_level: 4
  21. down_ratio: 4
  22. dcn_v2: False
  23. CenterNetHead:
  24. head_planes: 256
  25. regress_ltrb: False
  26. CenterNetPostProcess:
  27. max_per_img: 100
  28. regress_ltrb: False