masktrack-rcnn_mask-rcnn_r50_fpn_8xb1-12e_youtubevis2021.py 541 B

1234567891011121314151617
  1. _base_ = ['./masktrack-rcnn_mask-rcnn_r50_fpn_8xb1-12e_youtubevis2019.py']
  2. data_root = 'data/youtube_vis_2021/'
  3. dataset_version = data_root[-5:-1]
  4. # dataloader
  5. train_dataloader = dict(
  6. dataset=dict(
  7. data_root=data_root,
  8. dataset_version=dataset_version,
  9. ann_file='annotations/youtube_vis_2021_train.json'))
  10. val_dataloader = dict(
  11. dataset=dict(
  12. data_root=data_root,
  13. dataset_version=dataset_version,
  14. ann_file='annotations/youtube_vis_2021_valid.json'))
  15. test_dataloader = val_dataloader