uni/ccv/saliency/main.cpp

#include <cstdio>
#include <opencv2/opencv.hpp>

#include "includes/lab_pyramid.h"
#include "includes/fusion.h"

/**
 * Entry point of program
 * @param argc number of arguments
 * @param argv CLI arguments, 0: name of program, 1: input file name, 2: output file name
 * @return status code (0: everything OK, -1: not right amount of arguments)
 */
int main(int argc, char** argv) {
  if (argc != 3) {
    printf("usage: <input file name> <output file name>\n");
    return -1;
  }

  // read image
  cv::Mat image = cv::imread(argv[1], cv::ImreadModes::IMREAD_COLOR);

  // tweakable factors
  int layers = 4;
  float sigma_center = 5;
  float sigma_surround = 9;

  // create LAB pyramids
  lab_pyramid lab_pyr_center = lab_pyramid(image);
  lab_pyr_center.create_pyramids(sigma_center, layers);
  lab_pyramid lab_pyr_surround = lab_pyramid(image);
  lab_pyr_surround.create_pyramids(sigma_surround, layers);

  // create contrast maps
  lab_pyramid::compute_dog(lab_pyr_center, lab_pyr_surround, layers);
  // create feature maps
  lab_pyramid::compute_feature_maps();
  // create conspicuity maps
  lab_pyramid::compute_conspicuity_maps();
  // get conspicuity maps
  cv::Mat C_l = lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_L);
  cv::Mat C_a = lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_A);
  cv::Mat C_b = lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_B);
  // get saliency map
  cv::Mat saliency = max_fusion_saliency(C_l, C_a, C_b);

  // convert saliency map to correct output format
  cv::Mat output_image;
  saliency.convertTo(output_image, CV_8UC1);

  // write output
  cv::imwrite(argv[2], output_image);

  return 0;
}