mask-rcnn_r50_fpg_crop640-50e_coco.py 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. _base_ = 'mask-rcnn_r50_fpn_crop640-50e_coco.py'
  2. norm_cfg = dict(type='BN', requires_grad=True)
  3. model = dict(
  4. neck=dict(
  5. type='FPG',
  6. in_channels=[256, 512, 1024, 2048],
  7. out_channels=256,
  8. inter_channels=256,
  9. num_outs=5,
  10. stack_times=9,
  11. paths=['bu'] * 9,
  12. same_down_trans=None,
  13. same_up_trans=dict(
  14. type='conv',
  15. kernel_size=3,
  16. stride=2,
  17. padding=1,
  18. norm_cfg=norm_cfg,
  19. inplace=False,
  20. order=('act', 'conv', 'norm')),
  21. across_lateral_trans=dict(
  22. type='conv',
  23. kernel_size=1,
  24. norm_cfg=norm_cfg,
  25. inplace=False,
  26. order=('act', 'conv', 'norm')),
  27. across_down_trans=dict(
  28. type='interpolation_conv',
  29. mode='nearest',
  30. kernel_size=3,
  31. norm_cfg=norm_cfg,
  32. order=('act', 'conv', 'norm'),
  33. inplace=False),
  34. across_up_trans=None,
  35. across_skip_trans=dict(
  36. type='conv',
  37. kernel_size=1,
  38. norm_cfg=norm_cfg,
  39. inplace=False,
  40. order=('act', 'conv', 'norm')),
  41. output_trans=dict(
  42. type='last_conv',
  43. kernel_size=3,
  44. order=('act', 'conv', 'norm'),
  45. inplace=False),
  46. norm_cfg=norm_cfg,
  47. skip_inds=[(0, 1, 2, 3), (0, 1, 2), (0, 1), (0, ), ()]))