1
0
mirror of https://github.com/2martens/uni.git synced 2026-05-06 19:36:26 +02:00

[CCV] Implemented first version of saliency system

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2017-05-30 16:13:25 +02:00
parent 63e58af14a
commit ae24a8e256
8 changed files with 516 additions and 0 deletions

54
ccv/saliency/main.cpp Normal file
View File

@ -0,0 +1,54 @@
#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;
}