归一化tensor并画出热图

球场书生

已于 2024-04-08 14:05:05 修改

阅读量206

点赞数 2

文章标签：计算机视觉 python matplotlib ai

于 2024-03-25 10:53:17 首次发布

本文链接：https://blog.csdn.net/qq_41917697/article/details/137007734

版权

本文介绍了如何使用Python库numpy和matplotlib.pyplot对深度学习模型的中间输出进行可视化处理，通过计算归一化张量并显示为热力图来观察模型内部特征的分布。

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

有时候需要可视化观察一下模型中间输出的结果，写个博客简单记录一下，方便后面直接粘贴复用

import numpy as np
import matplotlib.pyplot as plt
#xq_out.shape [B,H,W]
normalized_tensor = (xq_out[0] - xq_out[0].min()) / (xq_out[0].max() - xq_out[0].min())
plt.imshow(normalized_tensor.cpu(), cmap='hot')
plt.colorbar()
plt.savefig('./heatmap_org2.png', bbox_inches='tight', pad_inches=0) #, dpi=600 调整分辨率