当前位置:   article > 正文

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

loss_kps

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()

 

可视化结果如下图所示:

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/从前慢现在也慢/article/detail/342786
推荐阅读
相关标签
  

闽ICP备14008679号