cnns模型matlab,MatConvNet: CNNs for MATLAB

Quick start

If you are new to MatConvNet, cut & paste the following code in a MATLAB window to try out MatConvNet. The code downloads and compiles MatConvNet, downloads a pre-trained CNN, and uses the latter to classify one of MATLAB stock images.

This example requires MATLAB to be interfaced to a C/C++ compiler (try mex -setup if you are unsure). Depending on your Internet connection speed, downloading the CNN model may require some time.

% Install and compile MatConvNet (needed once).

untar('http://www.vlfeat.org/matconvnet/download/matconvnet-1.0-beta25.tar.gz') ;

cd matconvnet-1.0-beta25

run matlab/vl_compilenn ;

% Download a pre-trained CNN from the web (needed once).

urlwrite(...

'http://www.vlfeat.org/matconvnet/models/imagenet-vgg-f.mat', ...

'imagenet-vgg-f.mat') ;

% Setup MatConvNet.

run matlab/vl_setupnn ;

% Load a model and upgrade it to MatConvNet current version.

net = load('imagenet-vgg-f.mat') ;

net = vl_simplenn_tidy(net) ;

% Obtain and preprocess an image.

im = imread('peppers.png') ;

im_ = single(im) ; % note: 255 range

im_ = imresize(im_, net.meta.normalization.imageSize(1:2)) ;

im_ = im_ - net.meta.normalization.averageImage ;

% Run the CNN.

res = vl_simplenn(net, im_) ;

% Show the classification result.

scores = squeeze(gather(res(end).x)) ;

[bestScore, best] = max(scores) ;

figure(1) ; clf ; imagesc(im) ;

title(sprintf('%s (%d), score %.3f',...

net.meta.classes.description{best}, best, bestScore)) ;