picodet_l_416_coco.yml 947 B

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647
  1. _BASE_: [
  2. '../../datasets/coco_detection.yml',
  3. '../../runtime.yml',
  4. '_base_/picodet_esnet.yml',
  5. '_base_/optimizer_300e.yml',
  6. '_base_/picodet_416_reader.yml',
  7. ]
  8. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ESNet_x1_25_pretrained.pdparams
  9. weights: output/picodet_l_416_coco/model_final
  10. find_unused_parameters: True
  11. use_ema: true
  12. cycle_epoch: 40
  13. snapshot_epoch: 10
  14. epoch: 250
  15. ESNet:
  16. scale: 1.25
  17. feature_maps: [4, 11, 14]
  18. act: hard_swish
  19. channel_ratio: [0.875, 0.5, 1.0, 0.625, 0.5, 0.75, 0.625, 0.625, 0.5, 0.625, 1.0, 0.625, 0.75]
  20. CSPPAN:
  21. out_channels: 160
  22. PicoHead:
  23. conv_feat:
  24. name: PicoFeat
  25. feat_in: 160
  26. feat_out: 160
  27. num_convs: 4
  28. num_fpn_stride: 4
  29. norm_type: bn
  30. share_cls_reg: True
  31. feat_in_chan: 160
  32. TrainReader:
  33. batch_size: 48
  34. LearningRate:
  35. base_lr: 0.3
  36. schedulers:
  37. - !CosineDecay
  38. max_epochs: 300
  39. - !LinearWarmup
  40. start_factor: 0.1
  41. steps: 300