|
import os |
|
from pathlib import Path |
|
|
|
import pytest |
|
|
|
from mmdet.apis import init_detector |
|
|
|
|
|
def test_init_detector(): |
|
project_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__))) |
|
project_dir = os.path.join(project_dir, '..') |
|
|
|
config_file = os.path.join( |
|
project_dir, 'configs/mask_rcnn/mask_rcnn_r50_fpn_1x_coco.py') |
|
|
|
|
|
cfg_options = dict( |
|
model=dict( |
|
backbone=dict( |
|
depth=18, |
|
init_cfg=dict( |
|
type='Pretrained', checkpoint='torchvision://resnet18')))) |
|
model = init_detector(config_file, device='cpu', cfg_options=cfg_options) |
|
|
|
|
|
config_path_object = Path(config_file) |
|
model = init_detector(config_path_object, device='cpu') |
|
|
|
|
|
with pytest.raises(TypeError): |
|
config_list = [config_file] |
|
model = init_detector(config_list) |
|
|