最近写论文,需要把特征图以热图的形式展示出来,因此写下这篇笔记:
需要用到的库为seaborn
具体代码如下:
import seanborn as sns
sns.set()
# 注意: your_data只能是二维矩阵,特征图为多通道时可以使用mean函数取通道上的平均。
ax = sns.heatmap(your_data, cmap="rainbow", xticklabels=False, yticklabels=False, cbar=True)
b = ax.get_figure()
b.savefig(your_save_path)
seaborn库是一个可视化数据的库,heatmap是该库中的一个用来绘制热力图的函数。
heatmap()函数中有很多参数,可以根据自己的绘图需要进行选择,以下是该函数的完整参数。
seaborn.heatmap(data, vmin=None, vmax=None, cmap=None, center=None, robust=False, annot=None, fmt='.2g', annot_kws=None, linewidths=0, linecolor='white', cbar=True, cbar_kws=None, cbar_ax=None, square=False, xticklabels='auto', yticklabels='auto',