retinanet_r50_fpn.yml 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. architecture: RetinaNet
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
  3. RetinaNet:
  4. backbone: ResNet
  5. neck: FPN
  6. head: RetinaHead
  7. ResNet:
  8. depth: 50
  9. variant: b
  10. norm_type: bn
  11. freeze_at: 0
  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. RetinaHead:
  21. conv_feat:
  22. name: RetinaFeat
  23. feat_in: 256
  24. feat_out: 256
  25. num_convs: 4
  26. norm_type: null
  27. use_dcn: false
  28. anchor_generator:
  29. name: RetinaAnchorGenerator
  30. octave_base_scale: 4
  31. scales_per_octave: 3
  32. aspect_ratios: [0.5, 1.0, 2.0]
  33. strides: [8.0, 16.0, 32.0, 64.0, 128.0]
  34. bbox_assigner:
  35. name: MaxIoUAssigner
  36. positive_overlap: 0.5
  37. negative_overlap: 0.4
  38. allow_low_quality: true
  39. loss_class:
  40. name: FocalLoss
  41. gamma: 2.0
  42. alpha: 0.25
  43. loss_weight: 1.0
  44. loss_bbox:
  45. name: SmoothL1Loss
  46. beta: 0.0
  47. loss_weight: 1.0
  48. nms:
  49. name: MultiClassNMS
  50. nms_top_k: 1000
  51. keep_top_k: 100
  52. score_threshold: 0.05
  53. nms_threshold: 0.5