ttfnet_darknet53.yml 615 B

1234567891011121314151617181920212223242526272829303132333435
  1. architecture: TTFNet
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/DarkNet53_pretrained.pdparams
  3. TTFNet:
  4. backbone: DarkNet
  5. neck: TTFFPN
  6. ttf_head: TTFHead
  7. post_process: BBoxPostProcess
  8. DarkNet:
  9. depth: 53
  10. freeze_at: 0
  11. return_idx: [1, 2, 3, 4]
  12. norm_type: bn
  13. norm_decay: 0.0004
  14. TTFFPN:
  15. planes: [256, 128, 64]
  16. shortcut_num: [3, 2, 1]
  17. TTFHead:
  18. hm_loss:
  19. name: CTFocalLoss
  20. loss_weight: 1.
  21. wh_loss:
  22. name: GIoULoss
  23. loss_weight: 5.
  24. reduction: sum
  25. BBoxPostProcess:
  26. decode:
  27. name: TTFBox
  28. max_per_img: 100
  29. score_thresh: 0.01
  30. down_ratio: 4