如何使用字符串按名称动态调用 Python 函数？

问题描述

我有一个像这样的 Python 函数调用：

import torchvision

model = torchvision.models.resnet18(pretrained=configs.use_trained_models)

哪个工作正常。

如果我试图让它变得动态：

import torchvision

model_name = 'resnet18'
model = torchvision.models[model_name](pretrained=configs.use_trained_models)

然后它失败了：

TypeError: 'module' object is not subscriptable

这是有道理的，因为 model 是一个导出一堆东西的模块，包括 resnet 函数：

# __init__.py for the "models" module

...
from .resnet import * 
...

如何在不提前知道函数名称的情况下动态调用这个函数（除了我得到一个带有函数名称的字符串）？

解决方法

您可以使用 getattr 函数：

import torchvision

model_name = 'resnet18'
model = getattr(torchvision.models,model_name)(pretrained=configs.use_trained_models)

这基本上与点符号相同，只是在函数形式中接受字符串来检索属性/方法。