Pytorch抽取网络层的抽取网络层的Feature Map(Vgg)实例实例
今天小编就为大家分享一篇Pytorch抽取网络层的Feature Map(Vgg)实例，具有很好的参考价值，希望对大家有
所帮助。一起跟随小编过来看看吧
这边我是需要得到图片在Vgg的5个block里relu后的Feature Map （其余网络只需要替换就可以了）
索引可以这样获得索引可以这样获得
vgg = models.vgg19(pretrained=True).features.eval()
print (vgg)
Feature Map可利用下面的可利用下面的class
class Vgg16(nn.Module):
def __init__(self, pretrained=True):
super(Vgg16, self).__init__()
self.net = models.vgg16(pretrained).features.eval()
def forward(self, x):
out = []
for i in range(len(self.net)):
x = self.net[i](x)
if i in [3, 8, 15, 22, 29]:
# print(self.net[i])
out.append(x
return out
以上这篇Pytorch抽取网络层的Feature Map(Vgg)实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大
家多多支持我们。
weixin_38556985
- 粉丝: 3
- 资源: 906
我的内容管理
收起
- 我的资源
快来上传第一个资源
我的收益 登录查看自己的收益
我的积分
登录查看自己的积分
我的C币
登录后查看C币余额
我的收藏 
我的下载 
下载帮助
会员权益专享
最新资源
- zigbee-cluster-library-specification
- JSBSim Reference Manual
- c++校园超市商品信息管理系统课程设计说明书(含源代码) (2).pdf
- 建筑供配电系统相关课件.pptx
- 企业管理规章制度及管理模式.doc
- vb打开摄像头.doc
- 云计算-可信计算中认证协议改进方案.pdf
- [详细完整版]单片机编程4.ppt
- c语言常用算法.pdf
- c++经典程序代码大全.pdf
- 单片机数字时钟资料.doc
- 11项目管理前沿1.0.pptx
- 基于ssm的“魅力”繁峙宣传网站的设计与实现论文.doc
- 智慧交通综合解决方案.pptx
- 建筑防潮设计-PowerPointPresentati.pptx
- SPC统计过程控制程序.pptx
资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！
点击此处反馈
评论0