picodet_s_320_coco_lcnet.yml 846 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '_base_/picodet_v2.yml',
  5. '_base_/optimizer_300e.yml',
  6. '_base_/picodet_320_reader.yml',
  7. ]
  8. pretrain_weights: https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPLCNet_x0_75_pretrained.pdparams
  9. weights: output/picodet_s_320_coco/best_model
  10. find_unused_parameters: True
  11. use_ema: true
  12. epoch: 300
  13. snapshot_epoch: 10
  14. LCNet:
  15. scale: 0.75
  16. feature_maps: [3, 4, 5]
  17. LCPAN:
  18. out_channels: 96
  19. PicoHeadV2:
  20. conv_feat:
  21. name: PicoFeat
  22. feat_in: 96
  23. feat_out: 96
  24. num_convs: 2
  25. num_fpn_stride: 4
  26. norm_type: bn
  27. share_cls_reg: True
  28. use_se: True
  29. feat_in_chan: 96
  30. TrainReader:
  31. batch_size: 64
  32. LearningRate:
  33. base_lr: 0.32
  34. schedulers:
  35. - !CosineDecay
  36. max_epochs: 300
  37. - !LinearWarmup
  38. start_factor: 0.1
  39. steps: 300