File size: 1,116 Bytes
6c9ac8f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
# Copyright (c) OpenMMLab. All rights reserved.
from unittest import TestCase
from mmengine.registry import MODELS
from parameterized import parameterized
from mmdet.testing import get_detector_cfg
from mmdet.utils import register_all_modules
register_all_modules()
class TestSemiBase(TestCase):
@parameterized.expand([
'soft_teacher/'
'soft-teacher_faster-rcnn_r50-caffe_fpn_180k_semi-0.1-coco.py',
])
def test_init(self, cfg_file):
model = get_detector_cfg(cfg_file)
# backbone convert to ResNet18
model.detector.backbone.depth = 18
model.detector.neck.in_channels = [64, 128, 256, 512]
model.detector.backbone.init_cfg = None
model = MODELS.build(model)
self.assertTrue(model.teacher.backbone)
self.assertTrue(model.teacher.neck)
self.assertTrue(model.teacher.rpn_head)
self.assertTrue(model.teacher.roi_head)
self.assertTrue(model.student.backbone)
self.assertTrue(model.student.neck)
self.assertTrue(model.student.rpn_head)
self.assertTrue(model.student.roi_head)
|