list_models¶
- torchvision.models.list_models(module: Optional[module] = None, include: Optional[Union[Iterable[str], str]] = None, exclude: Optional[Union[Iterable[str], str]] = None) list[str][原始碼]¶
返回一個包含已註冊模型名稱的列表。
- 引數:
module (ModuleType, optional) – 我們想要從中提取可用模型的模組。
include (str 或 Iterable[str], optional) – 用於從所有模型集中包含模型的過濾器。過濾器將傳遞給 fnmatch 以匹配 Unix shell 風格的萬用字元。如果存在多個過濾器，則結果是各個過濾器並集的模型。
exclude (str 或 Iterable[str], optional) – 在 include_filters 之後應用的過濾器，用於移除模型。過濾器將傳遞給 fnmatch 以匹配 Unix shell 風格的萬用字元。如果存在多個過濾器，則結果是移除所有匹配任何單個過濾器的模型。
- 返回:
一個包含可用模型名稱的列表。
- 返回型別:
models (list)
