mmengine.hub.get_model

mmengine.hub.get_model(cfg_path, pretrained=False, **kwargs)

Get built model from external package.

参数

• cfg_path (str) – External relative config path with prefix ‘package::’ and without suffix.

• pretrained (bool) – Whether to load pretrained model. Defaults to False.

• kwargs (dict) – Default arguments to build model.

实际案例

>>> model = get_model('mmdet::faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py', pretrained=True)
>>> type(model)
<class 'mmdet.models.detectors.faster_rcnn.FasterRCNN'>

返回

Built model.

返回类型

nn.Module

参数

• cfg_path (str) –

• pretrained (bool) –