报错解决方法 ImportError: cannot import name ‘ConvBNReLU‘ from ‘torchvision.models.mobilenetv2‘

今天在使用 torch 的 mobilenet 的时候遇到了报错：

ImportError: cannot import name 'ConvBNReLU' from 'torchvision.models.mobilenetv2'

这个是torch版本问题，新版修改了API，只需要将

from torchvision.models.mobilenetv2 import ConvBNReLU

修改为

from torchvision.models.mobilenetv2 import Conv2dNormActivation

然后将代码里的 ConvBNReLU 替换为 Conv2dNormActivation 即可。

可以看出这是 torch 版本更新时希望把模块的 norm 和 act 变得更灵活而做出的修改。