pafnet_lite.yml 870 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. architecture: TTFNet
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV3_large_x1_0_ssld_pretrained.pdparams
  3. norm_type: sync_bn
  4. TTFNet:
  5. backbone: MobileNetV3
  6. neck: TTFFPN
  7. ttf_head: TTFHead
  8. post_process: BBoxPostProcess
  9. MobileNetV3:
  10. scale: 1.0
  11. model_name: large
  12. feature_maps: [5, 8, 14, 17]
  13. with_extra_blocks: true
  14. lr_mult_list: [0.25, 0.25, 0.5, 0.5, 0.75]
  15. conv_decay: 0.00001
  16. norm_decay: 0.0
  17. extra_block_filters: []
  18. TTFFPN:
  19. planes: [96, 48, 24]
  20. shortcut_num: [2, 2, 1]
  21. lite_neck: true
  22. fusion_method: concat
  23. TTFHead:
  24. hm_head_planes: 48
  25. wh_head_planes: 24
  26. lite_head: true
  27. hm_loss:
  28. name: CTFocalLoss
  29. loss_weight: 1.
  30. wh_loss:
  31. name: GIoULoss
  32. loss_weight: 5.
  33. reduction: sum
  34. BBoxPostProcess:
  35. decode:
  36. name: TTFBox
  37. max_per_img: 100
  38. score_thresh: 0.01
  39. down_ratio: 4