mirror of
https://gitcode.com/gh_mirrors/ope/OpenFace.git
synced 2026-05-14 11:17:53 +00:00
Master commit of OpenFace.
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
%function test_example_CNN
|
||||
load mnist_uint8;
|
||||
|
||||
train_x = double(reshape(train_x',28,28,60000))/255;
|
||||
test_x = double(reshape(test_x',28,28,10000))/255;
|
||||
train_y = double(train_y');
|
||||
test_y = double(test_y');
|
||||
|
||||
%% ex1 Train a 6c-2s-12c-2s Convolutional neural network
|
||||
%will run 1 epoch in about 200 second and get around 11% error.
|
||||
%With 100 epochs you'll get around 1.2% error
|
||||
|
||||
rand('state',0)
|
||||
|
||||
cnn.layers = {
|
||||
struct('type', 'i') %input layer
|
||||
struct('type', 'c', 'outputmaps', 6, 'kernelsize', 5) %convolution layer
|
||||
struct('type', 's', 'scale', 2) %sub sampling layer
|
||||
struct('type', 'c', 'outputmaps', 12, 'kernelsize', 5) %convolution layer
|
||||
struct('type', 's', 'scale', 2) %subsampling layer
|
||||
};
|
||||
|
||||
|
||||
opts.alpha = 1;
|
||||
opts.batchsize = 50;
|
||||
opts.numepochs = 5;
|
||||
|
||||
cnn = cnnsetup(cnn, train_x, train_y);
|
||||
cnn = cnntrain(cnn, train_x, train_y, opts);
|
||||
|
||||
[er, bad] = cnntest(cnn, test_x, test_y);
|
||||
|
||||
%plot mean squared error
|
||||
figure; plot(cnn.rL);
|
||||
assert(er<0.12, 'Too big error');
|
||||
Reference in New Issue
Block a user