SpatiallySparseDatasetCIFAR100.cpp

#include "SpatiallySparseDatasetCIFAR100.h"
#include <vector>
#include <iostream>
#include <fstream>

void readCIFAR100File(std::vector<Picture *> &characters, const char *filename) {
  std::ifstream file(filename, std::ios::in | std::ios::binary);
  if (!file) {
    std::cout << "Cannot find " << filename << std::endl;
    exit(EXIT_FAILURE);
  }
  unsigned char label[2];
  while (file.read((char *)label, 2)) {
    OpenCVPicture *character = new OpenCVPicture(32, 32, 3, 128, label[1]);
    unsigned char bitmap[3072];
    float *matData = ((float *)(character->mat.data));
    file.read((char *)bitmap, 3072);
    for (int x = 0; x < 32; x++) {
      for (int y = 0; y < 32; y++) {
        for (int c = 0; c < 3; c++) {
          matData[y * 96 + x * 3 + (2 - c)] = bitmap[c * 1024 + y * 32 + x];
        }
      }
    }
    characters.push_back(character);
  }
  file.close();
}
SpatiallySparseDataset Cifar100TrainSet() {
  SpatiallySparseDataset dataset;
  dataset.name = "CIFAR-100 train set";
  dataset.type = TRAINBATCH;
  dataset.nFeatures = 3;
  dataset.nClasses = 100;
  char filenameTrain[] = "Data/CIFAR100/train.bin";
  readCIFAR100File(dataset.pictures, filenameTrain);
  return dataset;
}
SpatiallySparseDataset Cifar100TestSet() {
  SpatiallySparseDataset dataset;
  dataset.name = "CIFAR-100 test set";
  dataset.type = TESTBATCH;
  dataset.nFeatures = 3;
  dataset.nClasses = 100;
  char filenameTest[] = "Data/CIFAR100/test.bin";
  readCIFAR100File(dataset.pictures, filenameTest);
  return dataset;
}