faster-rcnn_r50-attn0010-dcn_fpn_1x_coco.py 575 B

12345678910111213141516
  1. _base_ = '../faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. plugins=[
  5. dict(
  6. cfg=dict(
  7. type='GeneralizedAttention',
  8. spatial_range=-1,
  9. num_heads=8,
  10. attention_type='0010',
  11. kv_stride=2),
  12. stages=(False, False, True, True),
  13. position='after_conv2')
  14. ],
  15. dcn=dict(type='DCN', deform_groups=1, fallback_on_stride=False),
  16. stage_with_dcn=(False, True, True, True)))