2martens/uni
1
0
Fork 0
mirror of https://github.com/2martens/uni.git synced 2026-05-06 11:26:25 +02:00
Code Releases Activity

[CCV] Finished main function

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2017-05-30 21:23:52 +02:00
parent 8e418a912e
commit 1ffc7f0241
1 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
ccv/saliency/main.cpp
View File

@ -8,24 +8,29 @@
/** /**
* Entry point of program * Entry point of program
* @param argc number of arguments * @param argc number of arguments
* @param argv CLI arguments, 0: name of program, 1: input file name, 2: output file name * @param argv CLI arguments, 0: name of program, 1: input file name, 2: output file name, 3: flag for usage of orientation
* @return status code (0: everything OK, -1: not right amount of arguments) * @return status code (0: everything OK, -1: not right amount of arguments)
*/ */
int main(int argc, char** argv) { int main(int argc, char** argv) {
if (argc != 3) { if (argc != 4) {
printf("usage: <input file name> <output file name>\n"); printf("usage: <input file name> <output file name> <use orientation>\n");
return -1; return -1;
} }
// read image // read image
cv::Mat image = cv::imread(argv[1], cv::ImreadModes::IMREAD_COLOR); cv::Mat image = cv::imread(argv[1], cv::ImreadModes::IMREAD_COLOR);
bool use_orientation = (strcmp(argv[3], "1") == 0);
// tweakable factors // tweakable factors
int layers = 4; int layers = 4;
float sigma_center = 5; float sigma_center = 3;
float sigma_surround = 9; float sigma_surround = 7;
float sigma_laplacian = 4; float sigma_laplacian = 4;
int number_orientations = 4; int number_orientations = 4;
int gabor_size = 7;
double wavelength = 5;
double standard_deviation = 5;
// create LAB pyramids // create LAB pyramids
lab_pyramid lab_pyr_center = lab_pyramid(image); lab_pyramid lab_pyr_center = lab_pyramid(image);
@ -36,22 +41,29 @@ int main(int argc, char** argv) {
gauss_pyramid c_pyr_L = lab_pyr_center.get_pyramid(lab_pyramid::COLOR_L); gauss_pyramid c_pyr_L = lab_pyr_center.get_pyramid(lab_pyramid::COLOR_L);
laplacian_pyramid laplacian_pyr = laplacian_pyramid(c_pyr_L, sigma_laplacian); laplacian_pyramid laplacian_pyr = laplacian_pyramid(c_pyr_L, sigma_laplacian);
// create orientation pyramid // create orientation pyramid
oriented_pyramid oriented_pyr = oriented_pyramid(laplacian_pyr, number_orientations); oriented_pyramid oriented_pyr = oriented_pyramid(laplacian_pyr, number_orientations, gabor_size, wavelength,
standard_deviation);
// create contrast maps // create contrast maps
lab_pyramid::compute_dog(lab_pyr_center, lab_pyr_surround, layers); lab_pyramid::compute_dog(lab_pyr_center, lab_pyr_surround, layers);
// create feature maps // create feature maps
lab_pyramid::compute_feature_maps(); lab_pyramid::compute_feature_maps();
oriented_pyr.compute_feature_maps(); if (use_orientation) {
oriented_pyr.compute_feature_maps();
}
// create conspicuity maps // create conspicuity maps
lab_pyramid::compute_conspicuity_maps(); lab_pyramid::compute_conspicuity_maps();
oriented_pyr.compute_conspicuity_map(); if (use_orientation) {
oriented_pyr.compute_conspicuity_map();
}
// get conspicuity maps // get conspicuity maps
std::vector<cv::Mat> conspicuity_maps = std::vector<cv::Mat>(); std::vector<cv::Mat> conspicuity_maps = std::vector<cv::Mat>();
conspicuity_maps.push_back(lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_L)); conspicuity_maps.push_back(lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_L));
conspicuity_maps.push_back(lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_A)); conspicuity_maps.push_back(lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_A));
conspicuity_maps.push_back(lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_B)); conspicuity_maps.push_back(lab_pyramid::get_conspicuity_map(lab_pyramid::COLOR_B));
conspicuity_maps.push_back(oriented_pyr.get_conspicuity_map()); if (use_orientation) {
conspicuity_maps.push_back(oriented_pyr.get_conspicuity_map());
}
// get saliency map // get saliency map
cv::Mat saliency = max_fusion_generic(conspicuity_maps); cv::Mat saliency = max_fusion_generic(conspicuity_maps);