pafnet.yml 722 B

1234567891011121314151617181920212223242526272829303132333435363738394041
  1. architecture: TTFNet
  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. TTFNet:
  7. backbone: ResNet
  8. neck: TTFFPN
  9. ttf_head: TTFHead
  10. post_process: BBoxPostProcess
  11. ResNet:
  12. depth: 50
  13. variant: d
  14. return_idx: [0, 1, 2, 3]
  15. freeze_at: -1
  16. norm_decay: 0.
  17. variant: d
  18. dcn_v2_stages: [1, 2, 3]
  19. TTFFPN:
  20. planes: [256, 128, 64]
  21. shortcut_num: [3, 2, 1]
  22. TTFHead:
  23. dcn_head: true
  24. hm_loss:
  25. name: CTFocalLoss
  26. loss_weight: 1.
  27. wh_loss:
  28. name: GIoULoss
  29. loss_weight: 5.
  30. reduction: sum
  31. BBoxPostProcess:
  32. decode:
  33. name: TTFBox
  34. max_per_img: 100
  35. score_thresh: 0.01
  36. down_ratio: 4