ppyoloe_r_crn.yml 959 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. architecture: YOLOv3
  2. norm_type: sync_bn
  3. use_ema: true
  4. ema_decay: 0.9998
  5. YOLOv3:
  6. backbone: CSPResNet
  7. neck: CustomCSPPAN
  8. yolo_head: PPYOLOERHead
  9. post_process: ~
  10. CSPResNet:
  11. layers: [3, 6, 6, 3]
  12. channels: [64, 128, 256, 512, 1024]
  13. return_idx: [1, 2, 3]
  14. use_large_stem: True
  15. use_alpha: True
  16. CustomCSPPAN:
  17. out_channels: [768, 384, 192]
  18. stage_num: 1
  19. block_num: 3
  20. act: 'swish'
  21. spp: true
  22. use_alpha: True
  23. PPYOLOERHead:
  24. fpn_strides: [32, 16, 8]
  25. grid_cell_offset: 0.5
  26. use_varifocal_loss: true
  27. static_assigner_epoch: -1
  28. loss_weight: {class: 1.0, iou: 2.5, dfl: 0.05}
  29. static_assigner:
  30. name: FCOSRAssigner
  31. factor: 12
  32. threshold: 0.23
  33. boundary: [[512, 10000], [256, 512], [-1, 256]]
  34. assigner:
  35. name: RotatedTaskAlignedAssigner
  36. topk: 13
  37. alpha: 1.0
  38. beta: 6.0
  39. nms:
  40. name: MultiClassNMS
  41. nms_top_k: 2000
  42. keep_top_k: -1
  43. score_threshold: 0.1
  44. nms_threshold: 0.1
  45. normalized: False