fcos_r50_fpn.yml 916 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. architecture: FCOS
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
  3. FCOS:
  4. backbone: ResNet
  5. neck: FPN
  6. fcos_head: FCOSHead
  7. ResNet:
  8. depth: 50
  9. variant: 'b'
  10. norm_type: bn
  11. freeze_at: 0 # res2
  12. return_idx: [1, 2, 3]
  13. num_stages: 4
  14. FPN:
  15. out_channel: 256
  16. spatial_scales: [0.125, 0.0625, 0.03125]
  17. extra_stage: 2
  18. has_extra_convs: True
  19. use_c5: False
  20. FCOSHead:
  21. fcos_feat:
  22. name: FCOSFeat
  23. feat_in: 256
  24. feat_out: 256
  25. num_convs: 4
  26. norm_type: "gn"
  27. use_dcn: False
  28. fpn_stride: [8, 16, 32, 64, 128]
  29. prior_prob: 0.01
  30. norm_reg_targets: True
  31. centerness_on_reg: True
  32. num_shift: 0.5
  33. fcos_loss:
  34. name: FCOSLoss
  35. loss_alpha: 0.25
  36. loss_gamma: 2.0
  37. iou_loss_type: "giou"
  38. reg_weights: 1.0
  39. nms:
  40. name: MultiClassNMS
  41. nms_top_k: 1000
  42. keep_top_k: 100
  43. score_threshold: 0.025
  44. nms_threshold: 0.6