把tensorboard的多条训练损失曲线画在一张图上

在做实验的时候，通常需要对比分析不同模型的训练损失情况，tensorboard画出来的通常是不同的损失在不同的图上，那如果我们想把不同模型的训练损失画在同一张图进行对比分析，应该怎么处理呢？

可以通过matplotlib模块进行操作。

一、下载Tensorboard中的loss曲线的数据（下载方式如图所示）

 

 

 

 

二、用matplotlib模块画图

 

 

import csv
import matplotlib.pyplot as plt
import pandas as pd

df1 = pd.read_csv('/home/azhihong/JL-DCF-pytorch/log/run16-07/07-16-training loss.csv')  # csv文件所在路径
step1 = df1['Step'].values.tolist()
loss1 = df1['Value'].values.tolist()

df2 = pd.read_csv('/home/azhihong/JL-DCF-pytorch/log/run17-08/08-17-training loss.csv')
step2 = df2['Step'].values.tolist()
loss2 = df2['Value'].values.tolist()

plt.plot(step1, loss1, label='JL_DCF')
plt.plot(step2, loss2, label='Ours')
plt.legend(fontsize=16)  # 图注的大小
plt.show()

 

结果入下图所示

完结！