mask-rcnn_swin-s-p4-w7_fpn_amp-ms-crop-3x_coco.py 317 B

123456
  1. _base_ = './mask-rcnn_swin-t-p4-w7_fpn_amp-ms-crop-3x_coco.py'
  2. pretrained = 'https://github.com/SwinTransformer/storage/releases/download/v1.0.0/swin_small_patch4_window7_224.pth' # noqa
  3. model = dict(
  4. backbone=dict(
  5. depths=[2, 2, 18, 2],
  6. init_cfg=dict(type='Pretrained', checkpoint=pretrained)))