picodet_s_320_coco.yml 757 B

12345678910111213141516171819202122232425262728293031323334
  1. _BASE_: [
  2. '../../datasets/coco_detection.yml',
  3. '../../runtime.yml',
  4. '_base_/picodet_esnet.yml',
  5. '_base_/optimizer_300e.yml',
  6. '_base_/picodet_320_reader.yml',
  7. ]
  8. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/ESNet_x0_75_pretrained.pdparams
  9. weights: output/picodet_s_320_coco/model_final
  10. find_unused_parameters: True
  11. use_ema: true
  12. cycle_epoch: 40
  13. snapshot_epoch: 10
  14. ESNet:
  15. scale: 0.75
  16. feature_maps: [4, 11, 14]
  17. act: hard_swish
  18. channel_ratio: [0.875, 0.5, 0.5, 0.5, 0.625, 0.5, 0.625, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5]
  19. CSPPAN:
  20. out_channels: 96
  21. PicoHead:
  22. conv_feat:
  23. name: PicoFeat
  24. feat_in: 96
  25. feat_out: 96
  26. num_convs: 2
  27. num_fpn_stride: 4
  28. norm_type: bn
  29. share_cls_reg: True
  30. feat_in_chan: 96