show_loss

import os
import matplotlib.pyplot as plt
import numpy as np

# f = open('log1.txt','r') # loss
f = open('log2.txt','r') # +dcn
lines = f.readlines()

epoch=[]
loss_cls=[]
loss_pts_init=[]
loss_pts_refine=[]
loss=[]

for line in lines:
    line = line.split()
    # print(line)
    # print(len(line)) 
    epoch1 = line[6].split('[')[1][:-1]
    l_cls = line[-7][:-1]
    l_pts_init = line[-5][:-1]
    l_pts_refine = line[-3][:-1]
    l = line[-1]
    epoch.append(epoch1)
    loss_cls.append(l_cls)
    loss_pts_init.append(l_pts_init)
    loss_pts_refine.append(l_pts_refine)
    loss.append(l)
    # print(epoch, loss_cls, loss_pts_init, loss_pts_refine, loss)
loss_cls = dict(zip(epoch,loss_cls))
loss_pts_init = dict(zip(epoch,loss_pts_init))
loss_pts_refine = dict(zip(epoch,loss_pts_refine))
loss = dict(zip(epoch,loss))

epoch = [int(x) for x in loss.keys()]
loss = [float(y) for y in loss.values()]
loss_cls = [float(y) for y in loss_cls.values()]
loss_pts_init = [float(y) for y in loss_pts_init.values()]
loss_pts_refine = [float(y) for y in loss_pts_refine.values()]


# print(epoch,loss)
plt.plot(epoch, loss, 'r', label='loss')
plt.plot(epoch, loss_cls, 'green', label='loss_cls')
plt.plot(epoch, loss_pts_init, 'cyan', label='loss_pts_init')
plt.plot(epoch, loss_pts_refine, 'b', label='loss_pts_refine')
plt.legend()
plt.xlabel('epoch')
plt.ylabel('loss')
plt.show()