Function read_image

tools/compute_mean/uniform_mean.cpp:41–51

* Load an image from a file of the given name and return a cv:Mat object. * The image file can either be in a standard format handled by OpenCV, * or be in a proprietary format (a binary dump of a cv::Mat). */

 * or be in a proprietary format (a binary dump of a cv::Mat).
 */
cv::Mat read_image(const std::string image_file_name) {
  cv::Mat image;
  std::string ext = lbann::get_ext_name(image_file_name);
  std::transform(ext.begin(), ext.end(), ext.begin(), ::tolower);
  if (ext == "bin") { // load the binary dump of cv::Mat data
    image = lbann::cv_subtractor::read_binary_image_file(image_file_name);
  } else { // let OpenCV handle
    image = cv::imread(image_file_name);
  }
  return image;
}


/// Return the Channel-wise mean (the mean pixel value) of an image