picodet_l_320_coco_lcnet.yml 848 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_x2_0_pretrained.pdparams
  9. weights: output/picodet_l_320_coco/best_model
  10. find_unused_parameters: True
  11. use_ema: true
  12. epoch: 250
  13. snapshot_epoch: 10
  14. LCNet:
  15. scale: 2.0
  16. feature_maps: [3, 4, 5]
  17. LCPAN:
  18. out_channels: 160
  19. PicoHeadV2:
  20. conv_feat:
  21. name: PicoFeat
  22. feat_in: 160
  23. feat_out: 160
  24. num_convs: 4
  25. num_fpn_stride: 4
  26. norm_type: bn
  27. share_cls_reg: True
  28. use_se: True
  29. feat_in_chan: 160
  30. LearningRate:
  31. base_lr: 0.12
  32. schedulers:
  33. - !CosineDecay
  34. max_epochs: 300
  35. - !LinearWarmup
  36. start_factor: 0.1
  37. steps: 300
  38. TrainReader:
  39. batch_size: 24