可视化学习笔记2-pytorch 可视化卷积网络中间特征层

李卓璐

已于 2022-05-18 14:49:50 修改

阅读量1.2k

点赞数 2

分类专栏：算法实战文章标签： pytorch 深度学习学习

于 2022-05-18 14:47:09 首次发布

本文链接：https://blog.csdn.net/LZL2020LZL/article/details/124842035

版权

算法实战专栏收录该内容

64 篇文章 79 订阅 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

本文为可视化学习笔记的第二部分，主要介绍如何使用PyTorch来展示卷积网络在训练过程中不同层的中间特征图。通过运行代码，可以观察到第6层和第14层的特征输出，帮助理解网络内部的工作机制。

摘要由CSDN通过智能技术生成

输出的结果是两张特征图（可以实现跑一次看多层），代码如下。

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import requests
import cv2
import torch
from torch import nn
import torch.nn.functional as F
from torchvision import models
from torchvision import transforms
from PIL import Image
from model_M7 import