qq_57606524的博客

原创 如何获取模型最后一个全连接层之前的特征值，或修改模型的部分结构

这里要根据自己的模型的结构来编写代码。我这里是model[1].module.classifier,将其最后一层替换。修改后的效果如下，可以看到成功去掉了位于model[1].module.classifier的6号层，也就是模型最后的全连接层。将模型转化为list,进行切片操作，去除最后一层，然后再转回nn.Sequential。这样就可以使用新模型backbone来获取全连接层之前的特征feature了。以一个修改后的vgg16模型结构为例。方法也可以用来替换模型的一部分层。