如何从 vgg19 中移除自适应平均池层？

Posted 2023-03-28

【中文标题】如何从 vgg19 中移除自适应平均池层？

【英文标题】：How to remove the adaptive average pool layer from vgg19?

【发布时间】：2020-04-03 18:03:42

【问题描述】：

我已经加载了vgg19的预训练模型。如何去掉分类器前面的自适应平均池层？

【问题讨论】：

您使用的是哪个框架？张量流、keras、pytorch？

@mrzo 在 pytorch 我想知道我应该怎么做？

【参考方案1】：

如果你使用 PyTorch 和 torchVision 模型，你可以像这样禁用最后一个 maxpool 层 -

nn.Sequential(*list(vgg.features._modules.values())[:-1])

这里所有的层都在一个数组中，不包括最后一个元素就可以了。