深度学习-根据日志画loss-acc折线图

根据训练后的log日志画训练的loss-acc折线图

import matplotlib.pyplot as pltl_n = 0a_n = 0loss_x = []loss_y = []acc_x = []acc_y = []with open("/home/hytz/Downloads/Yifan/caffe-face/v2_log.txt") as f:   #根据自己的目录做对应的修改    for line in f:        line = line.strip()        if len(line.split("accuracy = ")) == 2:            acc_y.append(float(line.split("accuracy = ")[1]))            acc_x.append(a_n)            a_n += 100   #根据test_interval调整        if len(line.split(" loss = ")) == 2:            loss_y.append(float(line.split(" loss = ")[1]))              loss_x.append(l_n)            l_n += 20   #根据display调整print len(loss_x)       plt.figure(figsize=(8,6))plt.plot(loss_x,loss_y,'',label="loss")plt.plot(acc_x,acc_y,'',label="acc")plt.title('loss_acc')plt.legend(loc='upper right')plt.xlabel('iter')plt.ylabel('')plt.grid(loss_x)plt.show()

效果如下：