plot loss curve from caffe log file.

plotloss.py

import re
import matplotlib
import pylab as plt
import numpy as np

s = open('default.log','r').read()
pattern = re.compile(r'iter (\d+): (\d+.\d+)')
grou = np.asarray(pattern.findall(s)).astype(np.float32)
plt.plot(grou[:,0],grou[:,1])
plt.title('title')
plt.ylabel('batchloss')
plt.xlabel('iteration')
plt.show()