利用python对accuracy.txt文件中的loss可视化

loss文件如下：

可视化代码如下：

id_a = []
    id_k = []
    id_loss = []
    keypoint_loss = []
    id_loss_global = []
    fig = plt.figure()
    plt.xlim(xmax = 100, xmin = 0) #x轴取值范围
    plt.ylim(ymax =300, ymin = 0)#y轴取值范围
    with codecs.open('F:aa.txt','r') as f:
        lines = f.readlines()
        # print(lines)
        for line in lines:
            if 'loss = 'in line:
                b = line.split('loss = ')
                p = b[1]
                id_a.append(float(p[:-2]) * 100)
                # print(p[:-2])
                c = b[0].split(' = ')
                h = c[1].split(',')[0]
                # print(h)
                # plt.plot(int(h) / 10000, float(p[:-2]) * 100,'ro')
                id_k.append(int(h) / 10000)
                # id_k.append(h)
            if 'loss_id = 'in line:
                b = line.split('loss_id = ')
                p = b[1].split(' (')
                # print(p[0])
                id_loss.append(float(p[0]) * 1000)
            if 'loss_id_global = 'in line:
                b = line.split('loss_id_global = ')
                p = b[1].split(' (')
                # print(p[0])
                id_loss_global.append(float(p[0]) * 100)
            if 'loss_kps_mask = 'in line:
                b = line.split('loss_kps_mask = ')
                p = b[1].split(' (')
                print(p[0])
                keypoint_loss.append(float(p[0]) * 100)
    plt.plot(id_k,keypoint_loss,color='r',linestyle='-',linewidth=1)
    plt.plot(id_k,id_loss,color='g',linestyle='-',linewidth=1)
    plt.plot(id_k,id_loss_global,color='b',linestyle='-',linewidth=1)
    plt.legend(['keypoint_loss','id_loss','id_loss_global'],loc='upper right')
    plt.show()

 

可视化结果如下图所示：