Handwriting-Recognizer/mnist_load.h at master · partha04patel/Handwriting-Recognizer · 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
#include <fstream>
#include <vector>
#include <iostream>
#include <Eigen/Dense>
#include "nnimage.h"

using namespace std;
using namespace Eigen;
/* Helper library
 */
class mnist_loader {
public:
    static vector<nnimage> load(string fname, int num) {
        cout << "OK WE STARTED" << endl;
        ifstream imgs("Data/t10k-images-idx3-ubyte", ios::binary);
        ifstream lbls("Data/t10k-labels-idx1-ubyte", ios::binary);

        vector<nnimage> images;

        int checksum = 0;
        int image_count = 0;
        int image_width = 0;
        int image_height = 0;

        int label_checksum = 0;
        int label_count = 0;

        imgs.read((char*)&checksum, sizeof(checksum));
        imgs.read((char*)&image_count, sizeof(image_count));
        imgs.read((char*)&image_width, sizeof(image_width));
        imgs.read((char*)&image_height, sizeof(image_height));


        lbls.read((char*)&label_checksum, sizeof(label_checksum));
        lbls.read((char*)&label_count, sizeof(label_count));

        for (int i = 0; i < num; i++) {
            MatrixXd data(784, 1);

            unsigned char label = 0;
            lbls.read((char*)&label, sizeof(label));

            for (int j = 0; j < 784; j++) {
                unsigned char c = 0;
                imgs.read((char*)&c, sizeof(c));
                data(j, 0) = double(c);
            }
            nnimage nnimg(data, (int)label);
            images.push_back(nnimg);
        }

        imgs.close();

        return images;
    }
};