faster_rcnn_r50_fpn_gn_2x_coco.yml 829 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. _BASE_: [
  2. '../datasets/coco_detection.yml',
  3. '../runtime.yml',
  4. '../faster_rcnn/_base_/optimizer_1x.yml',
  5. '../faster_rcnn/_base_/faster_rcnn_r50_fpn.yml',
  6. '../faster_rcnn/_base_/faster_fpn_reader.yml',
  7. ]
  8. weights: output/faster_rcnn_r50_fpn_gn_2x_coco/model_final
  9. FasterRCNN:
  10. backbone: ResNet
  11. neck: FPN
  12. rpn_head: RPNHead
  13. bbox_head: BBoxHead
  14. # post process
  15. bbox_post_process: BBoxPostProcess
  16. FPN:
  17. out_channel: 256
  18. norm_type: gn
  19. BBoxHead:
  20. head: XConvNormHead
  21. roi_extractor:
  22. resolution: 7
  23. sampling_ratio: 0
  24. aligned: True
  25. bbox_assigner: BBoxAssigner
  26. XConvNormHead:
  27. num_convs: 4
  28. out_channel: 1024
  29. norm_type: gn
  30. epoch: 24
  31. LearningRate:
  32. base_lr: 0.01
  33. schedulers:
  34. - !PiecewiseDecay
  35. gamma: 0.1
  36. milestones: [16, 22]
  37. - !LinearWarmup
  38. start_factor: 0.1
  39. steps: 1000