cascade_rcnn_r50_fpn_gn_2x_coco.yml 712 B

12345678910111213141516171819202122232425262728293031323334353637
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '../cascade_rcnn/_base_/optimizer_1x.yml',
  5. '../cascade_rcnn/_base_/cascade_rcnn_r50_fpn.yml',
  6. '../cascade_rcnn/_base_/cascade_fpn_reader.yml',
  7. ]
  8. weights: output/cascade_rcnn_r50_fpn_gn_2x_coco/model_final
  9. FPN:
  10. out_channel: 256
  11. norm_type: gn
  12. CascadeHead:
  13. head: CascadeXConvNormHead
  14. roi_extractor:
  15. resolution: 7
  16. sampling_ratio: 0
  17. aligned: True
  18. bbox_assigner: BBoxAssigner
  19. CascadeXConvNormHead:
  20. num_convs: 4
  21. out_channel: 1024
  22. norm_type: gn
  23. epoch: 24
  24. LearningRate:
  25. base_lr: 0.01
  26. schedulers:
  27. - !PiecewiseDecay
  28. gamma: 0.1
  29. milestones: [16, 22]
  30. - !LinearWarmup
  31. start_factor: 0.1
  32. steps: 1000