赞
踩
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()
可视化结果如下图所示:
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。