visualise2d_digits.m

% function visualise2d(nn, images, labels)

hold off;
nSamples = 60;

images = imagesTrain;
labels = labelsTrain;

colors = [
    0    0.4470    0.7410
    0.8500    0.3250    0.0980
    0.9290    0.6940    0.1250
    0.4940    0.1840    0.5560
    0.4660    0.6740    0.1880
    0.3010    0.7450    0.9330
    0.6350    0.0780    0.1840
    0.1       0.1       0.1
    0.32      0.12       0.6
    0.9       0.3       0.5
];
nColors = size(colors,1);

im = [-1200:30];
grid = zeros(1140, 1800, 3); % 54, -60  and -140 40
figure(4);
codeLayer = numel(nn.rbm)/2+1;
% Should this use RBMs with binary states etc?


for digit = 0:9
    x = images(:, labels==digit);
    x = x(:, 1:nSamples);
    X = nnfeedforward(nn, x);
%     figure(2);
%     s{i+1} = scatter(X{codeLayer}(1,:), X{codeLayer}(2,:), 'filled', 'MarkerFaceColor', colors(i+1,:));


    code = X{codeLayer};
    c = round(code)*10;
    c(1,:) = c(1,:) + 1400;
    c(2,:) = c(2,:) + 540;
    c
    for j = 1:nSamples
        if sum(sum(sum( grid( c(2,j):c(2,j)+27, c(1,j):c(1,j)+27, :)))) == 0;
            rgbDigit = reshape((x(:,j))*(1-colors(mod(digit, nColors)+1,:)), 28, 28, 3);
            grid(c(2,j):c(2,j)+27, c(1,j):c(1,j)+27, :) = rgbDigit;
        end
    end

    image(1-grid);
    colormap gray;
    axis equal; 
    axis off;
    hold on;
    pause(0.3)
end

savefig('mnist-2d-digits')
% [hleg, hobj, hout, mout] = legend('0', '1', '2', '3', '4', '5', '6', '7', '8', '9');
% set(hleg, 'FontSize', 25, 'Location', 'northwest');
% legend boxoff;
% set(hleg, 'Position', [-.010 0.2 hleg.Position([3 4])]);
% for i = 1:10
%     hobj(i).FontSize = 25;
%     hobj(10+i).Children.MarkerSize = 20;
% end

% end

%