Caffe_Matlab/ShowFilters.m at master · ngcthuong/Caffe_Matlab · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
caffe.reset_all();
caffe.set_mode_cpu();

% net_model = 'C:\caffe-windows\examples\GoogLeNet\googlenet_deploy_prob.prototxt';
% net_weights = 'C:\caffe-windows\examples\GoogLeNet\thinned_net.caffemodel';
% net_model = 'C:\caffe-windows\models\flower\deploy.prototxt';
% net_weights = 'C:\caffe-windows\models\flower\thinnednet.caffemodel';
net_model = 'C:\caffe-windows\models\bvlc_reference_caffenet\deploy.prototxt';
net_weights = 'C:\caffe-windows\models\bvlc_reference_caffenet\bvlc_alexnet.caffemodel';
% net_model = 'C:\caffe-windows\models\googlenet_bn\inception_bn_deploy.prototxt';
% net_weights = 'C:\caffe-windows\models\googlenet_bn\inception_bn_1200000.caffemodel';
net = caffe.Net(net_model,net_weights,'test');
nth_layer = net.layer_vec(net.name2layer_index('conv1'));%/7x7_s2
nth_layer_blob1_data = nth_layer.params(1).get_data();

sizeB = size(nth_layer_blob1_data);
GridL = ceil(sqrt(sizeB(4)));

scale = 4;
border = 2;
sizeB(1) = sizeB(1) * scale;
sizeB(2) = sizeB(2) * scale;

background = zeros((border+sizeB(1))*GridL+border,(border+sizeB(2))*GridL+border,sizeB(3));
minV = min(nth_layer_blob1_data(:));
maxV = max(nth_layer_blob1_data(:));
nth_layer_blob1_data = (nth_layer_blob1_data - minV) / (maxV - minV);

for i = 1:sizeB(4)
    x = ceil(i / GridL);
    y = mod(i - 1,GridL) + 1;
    patch = imresize(nth_layer_blob1_data(:,:,:,i),[sizeB(1) sizeB(2)],'nearest');
    patch = (patch - min(patch(:))) / (max(patch(:)) - min(patch(:)));
    background(border + (x-1)*(border+sizeB(1)) + 1 : x*(border+sizeB(1)),border + (y-1)*(border+sizeB(2)) + 1 : y*(border+sizeB(2)),:) = patch;
end;

figure(3);
imshow(background);