mmcls.apis.list_models

mmcls.apis.list_models(pattern=None)

List all models available in MMClassification.

参数

pattern (str | None) – A wildcard pattern to match model names.

返回

a list of model names.

返回类型

List[str]

使用示例

List all models:

>>> from mmcls import list_models
>>> print(list_models())

List ResNet-50 models on ImageNet-1k dataset:

>>> from mmcls import list_models
>>> print(list_models('resnet*in1k'))
['resnet50_8xb32_in1k',
 'resnet50_8xb32-fp16_in1k',
 'resnet50_8xb256-rsb-a1-600e_in1k',
 'resnet50_8xb256-rsb-a2-300e_in1k',
 'resnet50_8xb256-rsb-a3-100e_in1k']