生成Vgg16的中間層輸出

Question

如何在 pytorch 模型中生成 VGG 16 的中間層輸出（第三和第四 MaxPool 層輸出）。

pretrained_model = models.vgg16(pretrained=True)

Answer 1

您可以按照此處所述使用前向掛鈎： https ://discuss.pytorch.org/t/how-can-l-load-my-best-model-as-a-feature-extractor-evaluator/17254/6

或者，子類化 vgg16 模型並操縱其前向函數。