Method CreateTestData

examples/cpp/imagenet/ilsvrc12.h:249–283

}

void ILSVRC::CreateTestData(string image_list, string input_folder,
                            string output_folder) {
  std::vector<std::pair<string, int>> file_list;
  string image_file_name;
  string outfile = output_folder + "/test.bin";
  int label;
  std::ifstream image_list_file(image_list.c_str(), std::ios::in);
  while (image_list_file >> image_file_name >> label)
    file_list.push_back(std::make_pair(image_file_name, label));
  LOG(INFO) << "Total number of test images is " << file_list.size();
  size_t num_test_images = file_list.size();
  for (size_t imageid = 0; imageid < num_test_images; imageid++) {
    string path = input_folder + "/" + file_list[imageid].first;
    Tensor image = ReadImage(path);
    label = file_list[imageid].second;
    Tensor lb(Shape{1}, singa::kInt);
    lb.CopyDataFromHostPtr<int>(&label, 1);
    std::vector<Tensor> input;
    input.push_back(image);
    input.push_back(lb);
    string encoded_str = encoder->Encode(input);
    if (writer == nullptr) {
      writer = new BinFileWriter();
      writer->Open(outfile, kCreate);
    }
    writer->Write(path, encoded_str);
  }
  if (writer != nullptr) {
    writer->Flush();
    writer->Close();
    delete writer;
    writer = nullptr;
  }
  LOG(INFO) << "Write " << num_test_images << " images into " << outfile;
}

void ILSVRC::ReadMean(string path) {
  BinFileReader bfreader;