- 博客(1)
- 收藏
- 关注
原创 如何获取模型最后一个全连接层之前的特征值,或修改模型的部分结构
这里要根据自己的模型的结构来编写代码。我这里是model[1].module.classifier,将其最后一层替换。修改后的效果如下,可以看到成功去掉了位于model[1].module.classifier的6号层,也就是模型最后的全连接层。将模型转化为list,进行切片操作,去除最后一层,然后再转回nn.Sequential。这样就可以使用新模型backbone来获取全连接层之前的特征feature了。以一个修改后的vgg16模型结构为例。方法也可以用来替换模型的一部分层。
2023-12-11 22:31:41
105
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人