python动态导入模块——importlib

当在写代码时，我们希望能够根据传入的选项设置，如args.model来确定要导入使用的是哪个model.py文件，而不是一股脑地导入

这种时候就需要用上python的动态导入模块

比如此时文件结构为：

├── models
│   ├── __init__.py
│   ├── cycle_gan_model.py
│   └── pix2pix_model.py
└── train.py

假设此时参数的设置中有：

import argparse

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument('--model', type=str, default='cycle_gan', help='chooses which model to use. [cycle_gan | pix2pix | test | colorization]')
args = parser.parse_args()

此时如果想要根据传入的参数args.model的值决定导入哪个模块，则：

 

import importlib
model_filename = "models." + args.model + "_model"
# 根据指定的args.model来导入本地models文件夹中的相应args.model + '_model.py'模块
# 假设如果args.model = cycle_gan，则导入模块models.cycle_gan_model
modellib = importlib.import_module(model_filename)

 

然后就能够使用modellib来调用cycle_gan_model模块

等价于：

from models import cycle_gan_model as modellib