【发布时间】：2020-04-03 18:03:42
【问题描述】：
我已经加载了vgg19的预训练模型。如何去掉分类器前面的自适应平均池层？
【问题讨论】：
-
您使用的是哪个框架？张量流、keras、pytorch？
-
@mrzo 在 pytorch 我想知道我应该怎么做？
标签： deep-learning vision vgg-net
【发布时间】：2020-04-03 18:03:42
【问题描述】：
如果你使用 PyTorch 和 torchVision 模型，你可以像这样禁用最后一个 maxpool 层 -
nn.Sequential(*list(vgg.features._modules.values())[:-1])
这里所有的层都在一个数组中，不包括最后一个元素就可以了。
【讨论】：