fcosr_x50.yml 891 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344
  1. architecture: YOLOv3
  2. snapshot_epoch: 1
  3. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ResNeXt50_32x4d_pretrained.pdparams
  4. YOLOv3:
  5. backbone: ResNet
  6. neck: FPN
  7. yolo_head: FCOSRHead
  8. post_process: ~
  9. ResNet:
  10. depth: 50
  11. groups: 32
  12. base_width: 4
  13. variant: b
  14. norm_type: bn
  15. freeze_at: 0
  16. return_idx: [1,2,3]
  17. num_stages: 4
  18. FPN:
  19. out_channel: 256
  20. extra_stage: 2
  21. has_extra_convs: true
  22. use_c5: false
  23. relu_before_extra_convs: true
  24. FCOSRHead:
  25. feat_channels: 256
  26. fpn_strides: [8, 16, 32, 64, 128]
  27. stacked_convs: 4
  28. loss_weight: {class: 1.0, probiou: 1.0}
  29. assigner:
  30. name: FCOSRAssigner
  31. factor: 12
  32. threshold: 0.23
  33. boundary: [[-1, 64], [64, 128], [128, 256], [256, 512], [512, 100000000.0]]
  34. nms:
  35. name: MultiClassNMS
  36. nms_top_k: 2000
  37. keep_top_k: -1
  38. score_threshold: 0.1
  39. nms_threshold: 0.1
  40. normalized: False