masktrack-rcnn_mask-rcnn_x101_fpn_8xb1-12e_youtubevis2021.py 1.1 KB

1234567891011121314151617181920212223242526272829303132
  1. _base_ = ['./masktrack-rcnn_mask-rcnn_r50_fpn_8xb1-12e_youtubevis2019.py']
  2. model = dict(
  3. detector=dict(
  4. backbone=dict(
  5. type='ResNeXt',
  6. depth=101,
  7. groups=64,
  8. base_width=4,
  9. init_cfg=dict(
  10. type='Pretrained',
  11. checkpoint='open-mmlab://resnext101_64x4d')),
  12. init_cfg=dict(
  13. type='Pretrained',
  14. checkpoint= # noqa: E251
  15. 'https://download.openmmlab.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_x101_64x4d_fpn_1x_coco/mask_rcnn_x101_64x4d_fpn_1x_coco_20200201-9352eb0d.pth' # noqa: E501
  16. )))
  17. data_root = 'data/youtube_vis_2021/'
  18. dataset_version = data_root[-5:-1]
  19. # dataloader
  20. train_dataloader = dict(
  21. dataset=dict(
  22. data_root=data_root,
  23. dataset_version=dataset_version,
  24. ann_file='annotations/youtube_vis_2021_train.json'))
  25. val_dataloader = dict(
  26. dataset=dict(
  27. data_root=data_root,
  28. dataset_version=dataset_version,
  29. ann_file='annotations/youtube_vis_2021_valid.json'))
  30. test_dataloader = val_dataloader