先跳转到tensorboard的曲线页面
cd logpth
tensorboard --logdir=unet
下载得到csv文件
import csv
import matplotlib.pyplot as plt
import pandas as pd
df1 = pd.read_csv(r'C:\Users\admin\Downloads/our_eval.csv') # csv文件所在路径
step1 = df1['Step'].values.tolist()
loss1 = df1['Value'].values.tolist()
df2 = pd.read_csv(r'C:\Users\admin\Downloads/unet_eval.csv')
step2 = df2['Step'].values.tolist()
loss2 = df2['Value'].values.tolist()
plt.plot(step1, loss1, label='Ours')
plt.plot(step2, loss2, label='unet')
plt.legend(fontsize=8) # 图注的大小
plt.title('Dsc')
plt.show()
完