CV学习1:vgg特征图可视化

代码下载

参考
利用vgg预训练模型提取图像特征

调试复现

1.更改权重位置

在这里插入图片描述

2.可视化代码

导入cv2,将输出的特征转化为np.array()且dtype=np.uint8,具体代码如图
在这里插入图片描述

3.运行test_vgg16.py

出现错误
ValueError: Object arrays cannot be loaded when allow_pickle=False
解决方法:更改numpy版本为1.16.2
参考文章:https://blog.csdn.net/weixin_42096901/article/details/89855804

4.运行结果

conv1_2的shape是[2,224,224,64],batch为2,一共64层特征图,我这里只显示了第1层和前3层

在这里插入图片描述

原图
在这里插入图片描述

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
### 回答1: 可以使用以下代码来生成VGG特征:import torch import torchvision.models as models import torchvision.transforms as transforms from torch.autograd import Variable import matplotlib.pyplot as plt# 载入VGG16模型 model = models.vgg16(pretrained=True)# 读入一张片 img = Image.open('path_to_image')# 对片做预处理 transform = transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize( mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225] ) ]) img_t = transform(img)# 把片做成一个batch batch_t = torch.unsqueeze(img_t, 0)# 将VGG16模型设置为可训练 for parma in model.parameters(): parma.requires_grad = False # 为片送入模型 out = model(Variable(batch_t))# 可视化VGG特征 plt.figure(figsize=(14, 8)) for i in range(out.size()[1]): ax = plt.subplot(8, 8, i + 1) ax.imshow(out[0, i, :, :].data.numpy(), cmap='jet') ax.axis('off') plt.show() ### 回答2: 要生成一个pytorch可视化VGG特征,需要进行以下步骤: 1. 导入必要的库和模块: ```python import torch import torch.nn as nn import torchvision.models as models import matplotlib.pyplot as plt ``` 2. 定义VGG模型并加载预训练的权重: ```python vgg = models.vgg16(pretrained=True) ``` 3. 获取所需层的特征: ```python layer_name = 'features.29' # 这里选择VGG16的最后一个卷积层,可以根据需要调整layer_name layer = vgg.features._modules[layer_name] ``` 4. 创建一个向前传递的hook函数,用于获取特征输出: ```python target_output = None def hook_fn(module, input, output): global target_output target_output = output ``` 5. 注册hook函数到目标层: ```python layer.register_forward_hook(hook_fn) ``` 6. 前向传播输入并得到特征: ```python input_img = torch.randn(1, 3, 224, 224) # 这里随机生成一个输入像,可以根据需要调整 _ = vgg(input_img) # 触发前向传播并生成特征 ``` 7. 将特征可视化: ```python feature_map = target_output[0].detach().numpy() # 将特征从Tensor转换为NumPy数组 plt.imshow(feature_map) plt.show() ``` 这样就可以生成并可视化VGG模型中指定层的特征了。注意,代码中的像尺寸、层的选择和其他参数可以根据需要进行调整。 ### 回答3: 要生成PyTorch中VGG特征的代码,需要首先导入必要的库和模块。代码如下所示: ```python import torch import torchvision.models as models import cv2 import matplotlib.pyplot as plt # 加载预训练的VGG模型 vgg = models.vgg16(pretrained=True) vgg.eval() # 选择指定层级的特征 selected_layer = vgg.features[12] # 加载并预处理像 image = cv2.imread('image.jpg') image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB) image = cv2.resize(image, (224, 224)) image = torch.from_numpy(image.transpose((2, 0, 1))) image = image.float() image /= 255.0 image = torch.unsqueeze(image, 0) # 前向传播获取输出特征 output = selected_layer(image) # 转换输出特征格式 output = output.data.squeeze() output = output.numpy() # 可视化特征 plt.imshow(output, cmap='gray') plt.axis('off') plt.show() ``` 以上代码首先加载了预训练的VGG16模型,并选择了第13层特征vgg.features[12])作为可视化对象。然后,加载和预处理了待处理的像,将其转换为PyTorch张量。接下来,通过前向传播计算输出特征。最后,使用matplotlib库将特征可视化显示在屏幕上。请确保将像文件路径替换为实际像文件的路径。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

吾系桉宁

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值