[开发工具] vscode 远程调试绘图方法 python并画出loss曲线

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 开发工具 -> vscode 远程调试绘图方法 python并画出loss曲线 -> 正文阅读

[开发工具]vscode 远程调试绘图方法 python并画出loss曲线

通过 vscode 进行远程调试时无法直接显示远程绘图结果，每次通过保存图片的方式太过繁琐，其实可以通过显式配置 Linux 系统的 DISPLAY 环境变量实现直接显示绘图结果。

一、首先安装并配置 X11 安装包

sudo apt-get install xorg
sudo apt-get install openbox

并修改配置

vim /etc/ssh/sshd_config

X11Forwarding yes
X11DisplayOffset 10
PrintMotd no
PrintLastLog yes
TCPKeepAlive yes

ubuntu16.04的话，默认安装后是配置好的，不用改

二、通过远程连接工具查看 DISPLAY 环境变量的值

echo $DISPLAY
localhost:10.0

三、在 vscode 窗口终端中指定 DISPLAY 环境变量

export DISPLAY=localhost:10.0

这样就可以直接绘图了。

下面举例画出训练paddleocr中的db模型时候的loss曲线

import numpy as np
import matplotlib.pyplot as plt
import pylab as pl
from mpl_toolkits.axes_grid1.inset_locator import inset_axes

with open("output/db_mv3/train_0406.log", "r") as f:
    x = []
    y = []
    count = 0
    for line in f.readlines():
        line = line.strip('\n')  #去掉列表中每一个元素的换行符
        if "iter:" in line:
            count += 1
            if count%100==0:
                items = line.split(",")
                iter = float(items[1].strip(" ").split(" ")[1]) #iter
                x.append(iter)  
                #因为loss太小，所以放大方便看         
                loss = float(items[3].strip(" ").split(" ")[1])*10000
                y.append(loss)
            
    print("x:", x)
    #print("y:", y)
            

fig = plt.figure()       #figsize是图片的大小`
ax1 = fig.add_subplot(1, 1, 1) # ax1是子图的名字`

pl.plot(x,y,'g-',label=u'DB_MV3_0406')
# ‘’g‘’代表“green”,表示画出的曲线是绿色，“-”代表画的曲线是实线，可自行选择，label代表的是图例的名称，一般要在名称前面加一个u，如果名称是中文，会显示不出来，目前还不知道怎么解决。
pl.legend()
#显示图例
pl.xlabel(u'iters')
pl.ylabel(u'loss')
plt.title('Compare loss for different models in training')