VGG-16可视化卷积层

最新推荐文章于 2024-03-06 15:24:35 发布

可我不想做饼干

最新推荐文章于 2024-03-06 15:24:35 发布

阅读量158

点赞数

文章标签：深度学习人工智能

本文链接：https://blog.csdn.net/weixin_63403986/article/details/131993225

版权

import torch
import torchvision.models as models
import torchvision.transforms as transforms
from PIL import Image
import matplotlib.pyplot as plt

# 加载预训练的VGG-16模型
model = models.vgg16(pretrained=True)

# # 选择要可视化的卷积层
# layer_name = 'features.0'  # 示例中选择VGG-16的第一个卷积层
#
# # 获取指定卷积层的输出
# target_layer = model.features._modules[layer_name]

# 选择要可视化的卷积层
layer_index = 0

# 获取指定卷积层的输出
target_layer = model.features[layer_index]


# 读取并预处理输入图像
# img_path = "C:/Users/86152/Pictures/Camera Roll/p1.png"
img_path = "C:/Users/86152/Pictures/Camera Roll/15168016777396602.png"

img = Image.open(img_path)
preprocess = transforms.Compose([
    transforms.Resize((224, 224)),  # 调整大小为224x224像素
    transforms.ToTensor(),  # 转换为张量
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # 标准化
])
input_tensor = preprocess(img)
input_batch = input_tensor.unsqueeze(0)

# 将模型设置为评估模式并进行前向传播
model.eval()
with torch.no_grad():
    features = target_layer(input_batch)

# 可视化卷积层的特征图
for i in range(features.size(1)):
    plt.imshow(features[0, i].cpu(), cmap='viridis')
    plt.title('Feature Map ' + str(i))
    plt.show()