In [11]:
%load_ext autoreload
%autoreload 2
import cPickle as pickle
import os; import sys; sys.path.append('..')
import gp
import gp.nets as nets
from nolearn.lasagne.visualize import plot_loss
from nolearn.lasagne.visualize import plot_conv_weights
from nolearn.lasagne.visualize import plot_conv_activity
from nolearn.lasagne.visualize import plot_occlusion
from matplotlib.pyplot import imshow
import matplotlib.pyplot as plt
%matplotlib inline
In [2]:
PATCH_PATH = ('cylinder2_rgb_small')
In [3]:
X_train, y_train, X_test, y_test = gp.Patch.load_rgb(PATCH_PATH)
In [4]:
gp.Util.view_rgba(X_train[100], y_train[100])
In [5]:
cnn = nets.RGBNetPlusPlus()
In [6]:
cnn = cnn.fit(X_train, y_train)
In [7]:
test_accuracy = cnn.score(X_test, y_test)
In [8]:
test_accuracy
Out[8]:
In [9]:
plot_loss(cnn)
Out[9]:
In [12]:
# store CNN
sys.setrecursionlimit(1000000000)
with open(os.path.expanduser('~/Projects/gp/nets/RGBPlusPlus.p'), 'wb') as f:
pickle.dump(cnn, f, -1)
In [ ]: