1添加loss到txt文件

a=[]
for i in range(2):a.append(i)
file=open('data.txt','w')
file.write(a)
file.close()

2从txt中读取并画图

list = []
with open('data.txt', 'r') as f:for line in f.readlines():arr = line.split(',')
print(arr.__len__())
arr[0]=arr[0][1:]
arr[-1]=arr[-1][0:-2]
print(arr[0])
print(arr[-1])
for i in range(len(arr)):arr[i]=float(arr[i])
import matplotlib.pyplot as plt
x=[]
for i in range(len(arr)):x.append(i*100)
plt.plot(x,arr,label="ssd-loss")
plt.legend()
plt.show()

即可ok