【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()