masktrack-rcnn_mask-rcnn_r101_fpn_8xb1-12e_youtubevis2021.py 988 B

12345678910111213141516171819202122232425262728
  1. _base_ = ['./masktrack-rcnn_mask-rcnn_r50_fpn_8xb1-12e_youtubevis2019.py']
  2. model = dict(
  3. detector=dict(
  4. backbone=dict(
  5. depth=101,
  6. init_cfg=dict(
  7. type='Pretrained', checkpoint='torchvision://resnet101')),
  8. init_cfg=dict(
  9. type='Pretrained',
  10. checkpoint= # noqa: E251
  11. 'https://download.openmmlab.com/mmdetection/v2.0/mask_rcnn/mask_rcnn_r101_fpn_1x_coco/mask_rcnn_r101_fpn_1x_coco_20200204-1efe0ed5.pth' # noqa: E501
  12. )))
  13. data_root = 'data/youtube_vis_2021/'
  14. dataset_version = data_root[-5:-1]
  15. # dataloader
  16. train_dataloader = dict(
  17. dataset=dict(
  18. data_root=data_root,
  19. dataset_version=dataset_version,
  20. ann_file='annotations/youtube_vis_2021_train.json'))
  21. val_dataloader = dict(
  22. dataset=dict(
  23. data_root=data_root,
  24. dataset_version=dataset_version,
  25. ann_file='annotations/youtube_vis_2021_valid.json'))
  26. test_dataloader = val_dataloader