In [1]:
import os
os.chdir('/home/renwh/gaze/renwh/caffe_with_cudnnv3/')
import sys
sys.path.insert(0,'./python')
import caffe
from pylab import *
%matplotlib inline
caffe.set_device(0)
caffe.set_mode_gpu()
solver = caffe.SGDSolver('examples/mymodel/04/quick_solver.prototxt')
#You can choose to load your model status
#solver.restore('examples/mymodel/03/lenet_iter_1001.solverstate')
In [2]:
# each output is (batch size, feature dim, spatial dim)
[(k, v.data.shape) for k, v in solver.net.blobs.items()]
#[(k, v.data.shape) for k, v in solver.test_nets[0].blobs.items()]
Out[2]:
In [3]:
# just print the weight sizes (not biases)
[(k, v[0].data.shape) for k, v in solver.net.params.items()]
Out[3]:
In [4]:
solver.net.forward() # train net
solver.test_nets[0].forward() # test net (there can be more than one)
Out[4]:
In [5]:
# we use a little trick to tile the first eight images
imshow(solver.test_nets[0].blobs['data'].data[:8, 0].transpose(1, 0, 2).reshape(36, 8*60), cmap='gray')
print solver.net.blobs['label'].data[:8]
In [6]:
solver.step(1)
In [7]:
imshow(solver.net.params['conv1/5x5_s1'][0].diff[:, 0].reshape(8, 8, 5, 5)
.transpose(0, 2, 1, 3).reshape(8*5, 8*5), cmap='gray')
Out[7]:
Show the conv1 weights pics.
Then, I will train the model, and log some information.
In [8]:
%%time
niter = 1000
test_interval = 25
# losses will also be stored in the log
train_loss = zeros(niter)
mean_error= zeros(int(np.ceil(niter / test_interval)))
output = zeros((niter, 8, 3))
# the main solver loop
for it in range(niter):
solver.step(1) # SGD by Caffe
# store the train loss
train_loss[it] = solver.net.blobs['loss'].data
# store the output on the first test batch
# (start the forward pass at conv1 to avoid loading new data)
solver.test_nets[0].forward(start='conv1/5x5_s1')
output[it] = solver.test_nets[0].blobs['ip2'].data[:8]
if it % test_interval == 0:
# caculate the square error for each gaze vector
solver.test_nets[0].forward()
num_test = 100;
sub_error = zeros((num_test, 3))
square_error = zeros((num_test, 3))
sum_Euclidean_error = zeros(num_test)
for i in range(num_test):
sub_error[i,:] = np.subtract(solver.test_nets[0].blobs['gaze'].data[i]
, solver.test_nets[0].blobs['ip2'].data[i])
square_error = np.square(sub_error)
sum_Euclidean_error = np.sum(square_error,1)
sum_Euclidean_error = np.sqrt(sum_Euclidean_error)/6
mean_error[it // test_interval] = np.sum(sum_Euclidean_error,0)/num_test*180
print 'Iteration', it, '. Mean error is', mean_error[it // test_interval]
In [9]:
_, ax1 = subplots()
ax2 = ax1.twinx()
ax1.plot(arange(niter), train_loss)
ax2.plot(test_interval * arange(len(mean_error)), mean_error, 'r')
ax1.set_xlabel('iteration')
ax1.set_ylabel('train loss')
ax2.set_ylabel('mean error')
Out[9]:
**show you the train loss curve.
In [10]:
num_test = 1000
# (start the forward pass at conv1 to avoid loading new data)
solver.test_nets[0].forward(start='conv1/5x5_s1')
solver.test_nets[0].forward()
#figure(figsize=(10, 5))
#imshow(solver.test_nets[0].blobs['data'].data[:num_test, 0].transpose(1, 0, 2).reshape(36, num_test*60), cmap='gray')
#Print the label and train result
#for i in range(num_test):
# print solver.test_nets[0].blobs['label'].data[i,:3] ,'label<->ip2', solver.test_nets[0].blobs['ip2'].data[i]
print '--------------------------------------------------------------------------------------------------------------'
# caculate the square error for each gaze vector
sub_error = zeros((num_test, 3))
square_error = zeros((num_test, 3))
sum_square_error = zeros(num_test)
for i in range(num_test):
sub_error[i,:] = np.subtract(solver.test_nets[0].blobs['label'].data[i,:3], solver.test_nets[0].blobs['ip2'].data[i])
square_error = np.square(sub_error)
sum_square_error = np.sum(square_error,1)
sum_square_error = np.sqrt(sum_square_error)/6
#print sub_error[i,:],square_error[i,:],sum_square_error[i]
#print sum_square_error[i],
print num_test,'test pic, mean error is ',np.sum(sum_square_error,0)/num_test*180,'degree'
_, ax1 = subplots()
ax1.plot(arange(num_test), sum_square_error*180,'bo', label='sampled')
ax1.set_xlabel('num_test')
ax1.set_ylabel('sum_square_error')
Out[10]:
In [11]:
imshow(solver.net.params['conv1/5x5_s1'][0].diff[:, 0].reshape(8, 8, 5, 5)
.transpose(0, 2, 1, 3).reshape(8*5, 8*5), cmap='gray')
Out[11]:
In [13]:
figure(figsize=(10, 5))
imshow(solver.net.params['conv2/3x3_s1'][0].diff[:, 0].reshape(8, 8, 3, 3)
.transpose(0, 2, 1, 3).reshape(8*3, 8*3), cmap='gray')
Out[13]:
In [14]:
figure(figsize=(20, 10))
imshow(solver.test_nets[0].blobs['conv1/5x5_s1'].data[:8, :].reshape(16,32,32,56)
.transpose(0,2,1,3).reshape(16*32, 32*56), cmap='gray')
Out[14]:
In [17]:
figure(figsize=(20, 10))
imshow(solver.test_nets[0].blobs['conv2/3x3_s1'].data[:8, :].reshape(16, 32, 6, 12)
.transpose(0,2,1,3).reshape(16*6,32*12), cmap='gray')
Out[17]:
In [20]:
figure(figsize=(50, 25))
imshow(solver.test_nets[0].blobs['pool2/3x3_s2'].data[:8, :].reshape(64, 32, 8, 14)
.transpose(0,2,1,3).reshape(64*8,32*14), cmap='gray')
Out[20]:
In [21]:
figure(figsize=(100, 50))
imshow(solver.test_nets[0].blobs['inception_2/output'].data[:8, :].reshape(128, 32, 8, 14)
.transpose(0,2,1,3).reshape(128*8,32*14), cmap='gray')
Out[21]:
In [18]:
#solver.net.save('my_model.caffemodel') I do not know how to use this.
solver.snapshot() #SAVE MY MODEL IN THE DIR YOU DEFINE IN SOLVER FILE.