1 问题
我们使用PyTorch可视化工具-Tensorboard打开训练曲线如下;
当我们写论文时,下面的曲线图直接截图到文档里是不合适的,
①有背景阴影,难看;
②横纵坐标刻度值的字体太小,也没有标题;
总之直接截图放文档里,显得很不专业。
2 下载曲线数据
将曲线数据用表格的形式下载,然后用python自己绘制曲线图;
下载格式 .csv
训练损失曲线数据,用.csv格式导出,数据如下表所示。用python绘制出 step和Value的关系曲线
3 python自己绘制曲线
from matplotlib import pyplot as plt
import csv
import numpy as np
filename = "D:/python/pltt/train_loss.csv" #.csv文件路径,可以添加绝对路径
with open(filename) as f:
csvreader = csv.reader(f, delimiter = ",", quotechar='"')
for line in range(1): #1代表从文件第2行开始读取
next (csvreader)
print("lines: ",line) #打印行数
x = [] #横纵坐标分别建立了两个list
y = []
for row in csvreader:
x.append(float(row[1])) #读取第2、3列数据,放入list
y.append(float(row[2]))
# 可视化图线
plt.plot(x,y,'g', linewidth=1.0) #曲线颜色绿色,线条宽度1
plt.show()