solov2_r50_fpn.yml 720 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. architecture: SOLOv2
  2. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNet50_cos_pretrained.pdparams
  3. SOLOv2:
  4. backbone: ResNet
  5. neck: FPN
  6. solov2_head: SOLOv2Head
  7. mask_head: SOLOv2MaskHead
  8. ResNet:
  9. depth: 50
  10. freeze_at: 0
  11. return_idx: [0,1,2,3]
  12. num_stages: 4
  13. FPN:
  14. out_channel: 256
  15. SOLOv2Head:
  16. seg_feat_channels: 512
  17. stacked_convs: 4
  18. num_grids: [40, 36, 24, 16, 12]
  19. kernel_out_channels: 256
  20. solov2_loss: SOLOv2Loss
  21. mask_nms: MaskMatrixNMS
  22. SOLOv2MaskHead:
  23. mid_channels: 128
  24. out_channels: 256
  25. start_level: 0
  26. end_level: 3
  27. SOLOv2Loss:
  28. ins_loss_weight: 3.0
  29. focal_loss_gamma: 2.0
  30. focal_loss_alpha: 0.25
  31. MaskMatrixNMS:
  32. pre_nms_top_n: 500
  33. post_nms_top_n: 100