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']