备注

您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。

mmcls.apis.init_model

mmcls.apis.init_model(config, checkpoint=None, device='cuda:0', options=None)

Initialize a classifier from config file.

参数

• config (str or mmcv.Config) – Config file path or the config object.

• checkpoint (str, optional) – Checkpoint path. If left as None, the model will not load any weights.

• options (dict) – Options to override some settings in the used config.

返回

The constructed classifier.

返回类型

nn.Module