Method ReadFile

examples/cpp/cifar10/cifar10.h:59–81

  return;
}
const std::pair<Tensor, Tensor> Cifar10::ReadFile(string file) {
  Tensor images(Shape{kBatchSize, 3, kImageSize, kImageSize});
  Tensor labels(Shape{kBatchSize}, kInt);
  if (dir_path_.back() != '/') dir_path_.push_back('/');
  LOG(INFO) << "Reading file " << dir_path_ + file;
  std::ifstream data_file((dir_path_ + file).c_str(),
                          std::ios::in | std::ios::binary);
  CHECK(data_file.is_open()) << "Unable to open file " << dir_path_ + file;
  int label;
  char image[kImageVol];
  float float_image[kImageVol];
  int tmplabels[kBatchSize];
  for (size_t itemid = 0; itemid < kBatchSize; ++itemid) {
    // LOG(INFO) << "reading " << itemid << "-th image";
    ReadImage(&data_file, &label, image);
    for (size_t i = 0; i < kImageVol; i++)
      float_image[i] = static_cast<float>(static_cast<uint8_t>(image[i]));
    images.CopyDataFromHostPtr(float_image, kImageVol, itemid * kImageVol);
    tmplabels[itemid] = label;
  }
  labels.CopyDataFromHostPtr(tmplabels, kBatchSize);
  return std::make_pair(images, labels);
}

const std::pair<Tensor, Tensor> Cifar10::ReadTrainData() {
  Tensor images(Shape{kBatchSize * kTrainFiles, 3, kImageSize, kImageSize});