In [11]:
%load_ext autoreload
%autoreload 2
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.RGBNet()
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 [12]:
plot_loss(cnn)
Out[12]:
In [ ]: