【python】 —— 库：TensorboardX（pytorch中的可视化工具）

1. 网上教程
2. 知乎
3. 文档

使用积累

writer.add_figure( )

from tensorboardX import SummaryWriter
import numpy as np
from matplotlib import pyplot as plt

writer = SummaryWriter(log_dir='run_image')

fig = plt.figure(figsize=(10, 10))  # 画布宽长比例

for i in range(2):
    score = np.random.randint(0, 255, [50, 50])
    name = list(range(score.shape[0]))
    col = list(range(score.shape[1]))

    ax = fig.add_subplot(1,2,1+i)
    # ax2 = fig.add_subplot(1,10,10)
    im=ax.imshow(score,cmap='YlOrRd') #用cmap设置配色方案
    # im2 = ax2.imshow(score, cmap='YlOrRd')  # 用cmap设置配色方案
    ax.xaxis.set_ticks_position('top') #设置x轴刻度到上方
    ax.set_xticks(np.arange(len(col))) #设置x轴刻度
    ax.set_yticks(np.arange(len(name))) #设置y轴刻度
    ax.set_xticklabels(col) #设置x轴刻度标签
    ax.set_yticklabels(name) #设置y轴刻度标签

    # ax2.xaxis.set_ticks_position('top')  # 设置x轴刻度到上方
    # ax2.set_xticks(np.arange(len(col)))  # 设置x轴刻度
    # ax2.set_yticks(np.arange(len(name)))  # 设置y轴刻度
    # ax2.set_xticklabels(col)  # 设置x轴刻度标签
    # ax2.set_yticklabels(name)  # 设置y轴刻度标签

    # fig.colorbar(im,pad=0.03) #设置颜色条
    # ax.set_title("HAHA",x=0.2,y=1.034,fontsize=16) #设置标题以及其位置和字体大小
    plt.savefig('./plot'+'.png',format='png')

    writer.add_figure('fig',fig,global_step=i)

writer.close()