centernet_mbv1_140e_coco.yml 454 B

123456789101112131415161718192021
  1. _BASE_: [
  2. 'centernet_r50_140e_coco.yml'
  3. ]
  4. pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/MobileNetV1_pretrained.pdparams
  5. weights: output/centernet_mbv1_140e_coco/model_final
  6. CenterNet:
  7. backbone: MobileNet
  8. neck: CenterNetDLAFPN
  9. head: CenterNetHead
  10. post_process: CenterNetPostProcess
  11. MobileNet:
  12. scale: 1.
  13. with_extra_blocks: false
  14. extra_block_filters: []
  15. feature_maps: [3, 5, 11, 13]
  16. TrainReader:
  17. batch_size: 32