ml-naive-bayes-SVM/tableGeneratorRandomSampling.m at main · seanemery1/ml-naive-bayes-SVM · 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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
function dataRandom = tableGeneratorRandomSampling(directoryPath)
% Creates a representative training dataset that is a downsample of the
% original 150 images, approximately 99% Hole and 1% Not-Hole.

srcLocation =...
    dir(strcat(directoryPath, 'Location\*.mat'));
srcUnlabelled =...
    dir(strcat(directoryPath, 'Unlabelled\*.jpg'));

% Initializing variables
maxIndex = max(size(srcLocation)); % Size of index (150 training images)
count = 0; % Count of pixels, X & Y -> single position on the table
dataRandom = zeros(48000000, 6); % Pixel count is an overestimate

% Loop through all 150 unlabelled images/location matrices
for index = 1:maxIndex
    % Obtaining filenames
    filenameLocation = strcat(directoryPath, 'Location\',...
        srcLocation(index).name);
    filenameUnlabelled = strcat(directoryPath, 'Unlabelled\',...
        srcUnlabelled(index).name);

    % Reading files to memory
    location = load(filenameLocation);
    location = location.location;
    cdataUnlabelled = imread(filenameUnlabelled);

    % Obtaining image dimensions (n = # of rows, m = # of columns)
    [n, m] = size(location);

    % Loop through every pixel in the image
    for height = 1:n
        for width = 1:m
            % Generating a random integer between 1 and 15 (inclusive)
            rand = randi([1 15]);
            % If random integer = 1, sample the pixel
            if rand == 1
                % Iterate through count
                count = count + 1;
                if location(height,width) == 1
                    dataRandom(count, 1) = height;
                    dataRandom(count, 2) = width;
                    dataRandom(count, 3) =...
                        cdataUnlabelled(height, width, 1);
                    dataRandom(count, 4) =...
                        cdataUnlabelled(height, width, 2);
                    dataRandom(count, 5) =...
                        cdataUnlabelled(height, width, 3);
                    dataRandom(count, 6) = 1;
                else
                    dataRandom(count, 1) = height;
                    dataRandom(count, 2) = width;
                    dataRandom(count, 3) =...
                        cdataUnlabelled(height, width, 1);
                    dataRandom(count, 4) =...
                        cdataUnlabelled(height, width, 2);
                    dataRandom(count, 5) =...
                    cdataUnlabelled(height, width, 3);
                    dataRandom(count, 6) = -1;
                end
            end
        end
    end
end
% Remove any excess, empty rows
dataRandom(count+1:end,:) = [];

% Saving Results Matrix
filenameMatrix = sprintf('resultsRandom');
nameAndPath = strcat(directoryPath, filenameMatrix);
save(nameAndPath, 'dataRandom');
end