Function ReadLocationsFile

tensorflow/examples/multibox_detector/main.cc:54–74

Takes a file name, and loads a list of comma-separated box priors from it, one per line, and returns a vector of the values.

// Takes a file name, and loads a list of comma-separated box priors from it,
// one per line, and returns a vector of the values.
Status ReadLocationsFile(const string& file_name, std::vector<float>* result,
                         size_t* found_label_count) {
  std::ifstream file(file_name);
  if (!file) {
    return tensorflow::errors::NotFound("Labels file ", file_name,
                                        " not found.");
  }
  result->clear();
  string line;
  while (std::getline(file, line)) {
    std::vector<string> string_tokens = tensorflow::str_util::Split(line, ',');
    result->reserve(string_tokens.size());
    for (const string& string_token : string_tokens) {
      float number;
      CHECK(tensorflow::strings::safe_strtof(string_token, &number));
      result->push_back(number);
    }
  }
  *found_label_count = result->size();
  return Status::OK();
}

// Given an image file name, read in the data, try to decode it as an image,
// resize it to the requested size, and then scale the values as desired.