modified reading mnist data

sony · Apr 25, 2019 · 802f88a · 802f88a
1 parent dcf3357
commit 802f88a
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/examples/cpp/mnist_collection/mnist_data.hpp b/examples/cpp/mnist_collection/mnist_data.hpp
@@ -35,12 +35,12 @@ vector<vector<uint8_t>> read_images(const string &train,
 
   gzFile fp = gzopen((data_root + filename).c_str(), "rb");
   if (fp == NULL) {
-    cerr << "This sample requires mnist data downloaded before.";
+    cerr << "This sample requires mnist data downloaded before." << endl;
     exit(0);
   }
 
-  int header;
-  gzread(fp, (char *)&header, 4 * sizeof(int));
+  char header[16];
+  gzread(fp, header, 16);
   int num_images = 60000;
   if (train == "test")
     num_images = 10000;
@@ -69,12 +69,12 @@ vector<uint8_t> read_labels(const string &train, const string &data_root,
 
   gzFile fp = gzopen((data_root + filename).c_str(), "rb");
   if (fp == NULL) {
-    cerr << "This sample requires mnist data downloaded before. ";
+    cerr << "This sample requires mnist data downloaded before." << endl;
     exit(0);
   }
 
-  int header;
-  gzread(fp, (char *)&header, 2 * sizeof(int));
+  char header[8];
+  gzread(fp, header, 8);
   int num_images = 60000;
   if (train == "test")
     num_images = 10000;

diff --git a/examples/cpp/mnist_training/mnist_training.hpp b/examples/cpp/mnist_training/mnist_training.hpp
@@ -40,12 +40,12 @@ vector<vector<uint8_t>> read_images(const string &data_root,
 
   gzFile fp = gzopen((data_root + filename).c_str(), "rb");
   if (fp == NULL) {
-    cerr << "This sample requires mnist data downloaded before.";
+    cerr << "This sample requires mnist data downloaded before." << endl;
     exit(0);
   }
 
-  int header;
-  gzread(fp, (char *)&header, 4 * sizeof(int));
+  char header[16];
+  gzread(fp, header, 16);
   int num_images = 60000;
   int num_rows = 28;
   int num_cols = 28;
@@ -69,12 +69,12 @@ vector<uint8_t> read_labels(const string &data_root, const string &filename) {
 
   gzFile fp = gzopen((data_root + filename).c_str(), "rb");
   if (fp == NULL) {
-    cerr << "This sample requires mnist data downloaded before. ";
+    cerr << "This sample requires mnist data downloaded before." << endl;
     exit(0);
   }
 
-  int header;
-  gzread(fp, (char *)&header, 2 * sizeof(int));
+  char header[8];
+  gzread(fp, header, 8);
   int num_images = 60000;
 
   vector<uint8_t> labels;