function [problem, labels, numImagesAll] = computeLabels(dirDB, files)

%load mat

filesMat = dir([dirDB '*.mat']);

mat = load([dirDB filesMat(1).name]);

problem = mat.problem;

%keep only elements which are in 'files'

indexrem = [];

for p = 1 : numel(problem)

    %get filename from problem

    filenameP = problem(p).filename;

    %init

    findP = -1;

    %loop on files

    for f = 1 : numel(files)

        %get filename from files

        filenameF = files(f).name;

        %cmp

        if strcmp(filenameP, filenameF)

            findP = 1;

        end %if strcmp

    end %for f

    %if not found, remove

    if findP == -1

        indexrem = [indexrem p];

    end %if findP

end %for p

%remove indexrem

problem(indexrem) = [];

%get labels

labels = [problem.class];

%get num of images

numImagesAll = numel(problem);