1. 首页
  2. 编程语言
  3. PHP
  4. 使用pytorch实现可视化中间层的结果

使用pytorch实现可视化中间层的结果

上传者: 2020-09-21 10:49:04上传 PDF文件 106.48KB 热度 13次

一直比较想知道图片经过卷积之后中间层的结果,于是使用pytorch写了一个脚本查看,先看效果这是原图,随便从网上下载的一张大概224*224大小的图片,如下我们使用的VGG16,包含RULE层总共有30层可以可视化的结果,我们把这30层分别保存在30个文件夹中,每个文件中根据特征的大小保存了64~128张图片结果如下:原图大小为224224,经过第一层后大小为64224*224,下面是第一层可视化的结果,总共有64张这样的图片:这层的输出大小是 1128112*112,总共有128张这样的图片下面是完整的代码

用户评论