vgg特征图可视化

文章目录

• 代码下载
• 调试复现
• 1.更改权重位置
• 2.可视化代码
• 3.运行test_vgg16.py
• 4.运行结果

代码下载

参考
利用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层

原图