opencv - “顺序”对象没有属性“分类器”

Question

如何找到 Pytorch 模型的 in_features？model.classifier.in_features 在densenet121 上工作，但在vgg18 上工作，是否有任何功能可以在所有torchvision 模型上工作？

标签： opencvpytorchartificial-intelligenceconv-neural-networktorchvision

Answer 1

classifier是 的实现Sequential中的一个模块VGG，所以，如果你想访问in_features传递给 的classifier，你必须检查in_features第一层的。

models.vgg19().classifier[0].in_features

似乎不同的实现遵循不同的模式，因此确定in_features所有模型的最佳方法是直接检查源代码。