soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.01-coco.py 445 B

123456789
  1. _base_ = ['soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.1-coco.py']
  2. # 1% coco train2017 is set as labeled dataset
  3. labeled_dataset = _base_.labeled_dataset
  4. unlabeled_dataset = _base_.unlabeled_dataset
  5. labeled_dataset.ann_file = 'semi_anns/instances_train2017.1@1.json'
  6. unlabeled_dataset.ann_file = 'semi_anns/instances_train2017.1@1-unlabeled.json'
  7. train_dataloader = dict(
  8. dataset=dict(datasets=[labeled_dataset, unlabeled_dataset]))