用Python绘制训练曲线,python读取.csv表格数据绘制曲线,Tensorboard可视化训练曲线导出数据用Python自己绘制训练曲线,python绘制Excel表格曲线

文章目录

  • 1 问题
  • 2 下载曲线数据
  • 3 python自己绘制曲线

1 问题

我们使用PyTorch可视化工具-Tensorboard打开训练曲线如下;

当我们写论文时,下面的曲线图直接截图到文档里是不合适的,
①有背景阴影,难看;
②横纵坐标刻度值的字体太小,也没有标题;

总之直接截图放文档里,显得很不专业。
用Python绘制训练曲线,python读取.csv表格数据绘制曲线,Tensorboard可视化训练曲线导出数据用Python自己绘制训练曲线,python绘制Excel表格曲线_第1张图片

2 下载曲线数据

将曲线数据用表格的形式下载,然后用python自己绘制曲线图;
下载格式 .csv
用Python绘制训练曲线,python读取.csv表格数据绘制曲线,Tensorboard可视化训练曲线导出数据用Python自己绘制训练曲线,python绘制Excel表格曲线_第2张图片
训练损失曲线数据,用.csv格式导出,数据如下表所示。用python绘制出 step和Value的关系曲线

用Python绘制训练曲线,python读取.csv表格数据绘制曲线,Tensorboard可视化训练曲线导出数据用Python自己绘制训练曲线,python绘制Excel表格曲线_第3张图片

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]))   #读取第23列数据,放入list                                                            
        y.append(float(row[2]))  
    
    # 可视化图线                                                               
    plt.plot(x,y,'g', linewidth=1.0) #曲线颜色绿色,线条宽度1
                                            
plt.show()      

用Python绘制训练曲线,python读取.csv表格数据绘制曲线,Tensorboard可视化训练曲线导出数据用Python自己绘制训练曲线,python绘制Excel表格曲线_第4张图片

你可能感兴趣的:(论文格式必会的那些技能,深度学习,Python,plt.plot,python绘制.csv曲线,读取excel数据绘制曲线,Python绘制训练曲线,可视化Tensorboard)