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