1  clear
 2  close all
 3
 4
 5 addpath ./matlab
 6
 7 model= './models/bvlc_reference_caffenet/deploy.prototxt';
 8 weights= './models/bvlc_reference_caffenet/bvlc_reference_caffenet.caffemodel';
 9
10
11  net = caffe.Net(model, weights, 'test'); % create net and load weights
12
13 %% obtain params in diff layers and show
14 pdata = net.params('conv1',1).get_data();
15
16 vis_square(pdata,2,0.5);
17
18
19 net.blobs('data').reshape([227 227 3 1]);
20 net.reshape();
21
22 %% prepare the image
23 im_data = caffe.io.load_image('./examples/images/cat.jpg');
24  mean = load('./matlab/+caffe/imagenet/ilsvrc_2012_mean.mat');
25
26 %% subtract mean_data (already in W x H x C, BGR)
27  mean_data = mean.mean_data;
28  im_data = im_data - mean_data;
29
30  width = 227; height = 227;
31  im_data = imresize(im_data, [width, height]); % resize using Matlab's imresize
32  res = net.forward({im_data});
33
34  prob = res{1};
35
36 %% obtain features and show
37  ddata = net.blobs('conv2').get_data();
38  vis_square(ddata,2,0);

 1 function vis_square(data,padsize,padval)
 2
 3  data=net_data_normalize(data);
 4
 5 if ~ exist('padsize', 'var')
 6   padsize=1;
 7 end
 8 if ~ exist('padval', 'var')
 9   padval=0;
10 end
11 ndim=ndims(data);
12 % w*num*h*chanel
13 if ndim==4
14     fprintf('visualize params\n');
15     data=permute(data,[1,4,2,3]);
16 else ndim==3
17     fprintf('visualize maps\n');
18     data=permute(data,[1,3,2]);
19 end
20
21     n = (ceil(sqrt(size(data,2))));
22     data=padarray(data,[padsize n^2-size(data,2) padsize 0],'post');
23     data=reshape(data,size(data,1),n,n,size(data,3),size(data,4));
24     data=permute(data,[1,3,4,2,5]);
25     data=reshape(data,[size(data,1)*n,size(data,3)*n,size(data,5)]);
26
27 figure
28 if ndim==4
29     ;
30 else ndim==3
31     data=imrotate(data,-90);
32 end
33
34 imshow(imresize(data,[500,500],'nearest'))
35
36 end