根据特征图绘制热图

代码小白的成长

于 2023-02-08 12:18:54 发布

阅读量667

点赞数

分类专栏： pytorch 文章标签： python opencv matplotlib

本文链接：https://blog.csdn.net/weixin_43723625/article/details/128932497

版权

pytorch 专栏收录该内容

12 篇文章

订阅专栏

这段代码主要展示了如何利用Python中的cv2和seaborn库生成热力图。首先导入必要的库，然后对数据进行处理，方法一是使用seaborn的heatmap函数，而方法二是通过cv2的applyColorMap函数，配合matplotlib的imshow显示图像，并使用imwrite保存图片。注意cv2的颜色通道与matplotlib的通道顺序差异。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import cv2
        import numpy as np
        import seaborn as sns
        import matplotlib.pyplot as plt
        # x = gap
        for i in range(x.shape[1]):
            # # 方法1
            # ax = sns.heatmap(x[0][i].cpu().numpy())
            # # plt.show()
            # plt.close()
            # figure = ax.get_figure()
            # figure.savefig(r'C:\Users\kah\Desktop\DAM\sns_heatmap_{}.png'.format(i))

            # 方法二
            ax = cv2.applyColorMap(x[0][i].cpu().numpy().astype(np.uint8), cv2.COLORMAP_JET) #生成heat map
            # ax = ax[:, :, ::-1]  # 注意cv2（BGR）和matplotlib(RGB)通道是相反的
            plt.imshow(ax)
            cv2.imwrite(r'C:\Users\kah\Desktop\DAM\sns_heatmap_{}.png'.format(i), ax)
            # ax.savefig(r'C:\Users\kah\Desktop\DAM\sns_heatmap_{}.png'.format(i))