ppyoloe_plus_crn_t_p2_auxhead_300e_coco.yml 805 B

123456789101112131415161718192021222324252627282930313233343536
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. './_base_/optimizer_300e.yml',
  5. './_base_/ppyoloe_plus_crn_tiny_auxhead.yml',
  6. './_base_/ppyoloe_plus_tiny_reader.yml',
  7. ]
  8. log_iter: 100
  9. snapshot_epoch: 10
  10. weights: output/ppyoloe_plus_crn_t_p2_auxhead_300e_coco/model_final
  11. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/CSPResNetb_t_pretrained.pdparams
  12. depth_mult: 0.33
  13. width_mult: 0.375
  14. architecture: PPYOLOEWithAuxHead
  15. PPYOLOEWithAuxHead:
  16. backbone: CSPResNet
  17. neck: CustomCSPPAN
  18. yolo_head: PPYOLOEHead
  19. aux_head: SimpleConvHead
  20. post_process: ~
  21. CSPResNet:
  22. return_idx: [0, 1, 2, 3] # index 0 stands for P2
  23. CustomCSPPAN:
  24. out_channels: [384, 384, 384, 384]
  25. SimpleConvHead:
  26. fpn_strides: [32, 16, 8, 4]
  27. PPYOLOEHead:
  28. fpn_strides: [32, 16, 8, 4]