OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.9 KiB
83 lines
2.9 KiB
import os.path as osp |
|
import tempfile |
|
from copy import deepcopy |
|
|
|
import pytest |
|
from mmcv.utils import Config |
|
|
|
from mmdet.utils import replace_cfg_vals |
|
|
|
|
|
def test_replace_cfg_vals(): |
|
temp_file = tempfile.NamedTemporaryFile() |
|
cfg_path = f'{temp_file.name}.py' |
|
with open(cfg_path, 'w') as f: |
|
f.write('configs') |
|
|
|
ori_cfg_dict = dict() |
|
ori_cfg_dict['cfg_name'] = osp.basename(temp_file.name) |
|
ori_cfg_dict['work_dir'] = 'work_dirs/${cfg_name}/${percent}/${fold}' |
|
ori_cfg_dict['percent'] = 5 |
|
ori_cfg_dict['fold'] = 1 |
|
ori_cfg_dict['model_wrapper'] = dict( |
|
type='SoftTeacher', detector='${model}') |
|
ori_cfg_dict['model'] = dict( |
|
type='FasterRCNN', |
|
backbone=dict(type='ResNet'), |
|
neck=dict(type='FPN'), |
|
rpn_head=dict(type='RPNHead'), |
|
roi_head=dict(type='StandardRoIHead'), |
|
train_cfg=dict( |
|
rpn=dict( |
|
assigner=dict(type='MaxIoUAssigner'), |
|
sampler=dict(type='RandomSampler'), |
|
), |
|
rpn_proposal=dict(nms=dict(type='nms', iou_threshold=0.7)), |
|
rcnn=dict( |
|
assigner=dict(type='MaxIoUAssigner'), |
|
sampler=dict(type='RandomSampler'), |
|
), |
|
), |
|
test_cfg=dict( |
|
rpn=dict(nms=dict(type='nms', iou_threshold=0.7)), |
|
rcnn=dict(nms=dict(type='nms', iou_threshold=0.5)), |
|
), |
|
) |
|
ori_cfg_dict['iou_threshold'] = dict( |
|
rpn_proposal_nms='${model.train_cfg.rpn_proposal.nms.iou_threshold}', |
|
test_rpn_nms='${model.test_cfg.rpn.nms.iou_threshold}', |
|
test_rcnn_nms='${model.test_cfg.rcnn.nms.iou_threshold}', |
|
) |
|
|
|
ori_cfg_dict['str'] = 'Hello, world!' |
|
ori_cfg_dict['dict'] = {'Hello': 'world!'} |
|
ori_cfg_dict['list'] = [ |
|
'Hello, world!', |
|
] |
|
ori_cfg_dict['tuple'] = ('Hello, world!', ) |
|
ori_cfg_dict['test_str'] = 'xxx${str}xxx' |
|
|
|
ori_cfg = Config(ori_cfg_dict, filename=cfg_path) |
|
updated_cfg = replace_cfg_vals(deepcopy(ori_cfg)) |
|
|
|
assert updated_cfg.work_dir \ |
|
== f'work_dirs/{osp.basename(temp_file.name)}/5/1' |
|
assert updated_cfg.model.detector == ori_cfg.model |
|
assert updated_cfg.iou_threshold.rpn_proposal_nms \ |
|
== ori_cfg.model.train_cfg.rpn_proposal.nms.iou_threshold |
|
assert updated_cfg.test_str == 'xxxHello, world!xxx' |
|
ori_cfg_dict['test_dict'] = 'xxx${dict}xxx' |
|
ori_cfg_dict['test_list'] = 'xxx${list}xxx' |
|
ori_cfg_dict['test_tuple'] = 'xxx${tuple}xxx' |
|
with pytest.raises(AssertionError): |
|
cfg = deepcopy(ori_cfg) |
|
cfg['test_dict'] = 'xxx${dict}xxx' |
|
updated_cfg = replace_cfg_vals(cfg) |
|
with pytest.raises(AssertionError): |
|
cfg = deepcopy(ori_cfg) |
|
cfg['test_list'] = 'xxx${list}xxx' |
|
updated_cfg = replace_cfg_vals(cfg) |
|
with pytest.raises(AssertionError): |
|
cfg = deepcopy(ori_cfg) |
|
cfg['test_tuple'] = 'xxx${tuple}xxx' |
|
updated_cfg = replace_cfg_vals(cfg)
|
|
|