centernet_mbv3_small_140e_coco.yml 579 B

12345678910111213141516171819202122232425262728
  1. _BASE_: [
  2. 'centernet_r50_140e_coco.yml'
  3. ]
  4. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV3_small_x1_0_ssld_pretrained.pdparams
  5. weights: output/centernet_mbv3_small_140e_coco/model_final
  6. CenterNet:
  7. backbone: MobileNetV3
  8. neck: CenterNetDLAFPN
  9. head: CenterNetHead
  10. post_process: CenterNetPostProcess
  11. MobileNetV3:
  12. model_name: small
  13. scale: 1.
  14. with_extra_blocks: false
  15. extra_block_filters: []
  16. feature_maps: [4, 9, 12]
  17. CenterNetDLAFPN:
  18. first_level: 0
  19. last_level: 3
  20. down_ratio: 8
  21. dcn_v2: False
  22. TrainReader:
  23. batch_size: 32