1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118
|
//! \example tutorial-contrast-sharpening.cpp
#include <cstdlib>
#include <iostream>
#include <visp3/core/vpImage.h>
#include <visp3/gui/vpDisplayGDI.h>
#include <visp3/gui/vpDisplayOpenCV.h>
#include <visp3/gui/vpDisplayX.h>
#include <visp3/io/vpImageIo.h>
#if defined(VISP_HAVE_MODULE_IMGPROC)
//! [Include]
#include <visp3/imgproc/vpImgproc.h>
//! [Include]
#endif
int main(int argc, const char **argv)
{
//! [Macro defined]
#if defined(VISP_HAVE_MODULE_IMGPROC) && (defined(VISP_HAVE_X11) || defined(VISP_HAVE_GDI) || defined(VISP_HAVE_OPENCV))
//! [Macro defined]
//!
std::string input_filename = "Crayfish-low-contrast.png";
int blockRadius = 150;
int bins = 256;
float slope = 3.0f;
float sigma = 2.0f;
double weight = 0.5;
for (int i = 1; i < argc; i++) {
if (std::string(argv[i]) == "--input" && i + 1 < argc) {
input_filename = std::string(argv[i + 1]);
} else if (std::string(argv[i]) == "--blockRadius" && i + 1 < argc) {
blockRadius = atoi(argv[i + 1]);
} else if (std::string(argv[i]) == "--bins" && i + 1 < argc) {
bins = atoi(argv[i + 1]);
} else if (std::string(argv[i]) == "--slope" && i + 1 < argc) {
slope = (float)atof(argv[i + 1]);
} else if (std::string(argv[i]) == "--sigma" && i + 1 < argc) {
sigma = (float)atof(argv[i + 1]);
} else if (std::string(argv[i]) == "--weight" && i + 1 < argc) {
weight = atof(argv[i + 1]);
} else if (std::string(argv[i]) == "--help" || std::string(argv[i]) == "-h") {
std::cout << "Usage: " << argv[0]
<< " [--input <input image>]"
" [--blockRadius <block radius for CLAHE>] "
" [--bins <nb histogram bins for CLAHE>] [--slope <slope for CLAHE>]"
" [--sigma <Gaussian kernel standard deviation>] [--weight <unsharp mask weighting>]"
" [--help] [-h]"
<< std::endl;
return EXIT_SUCCESS;
}
}
//! [Read]
vpImage<vpRGBa> I_color;
vpImageIo::read(I_color, input_filename);
//! [Read]
#ifdef VISP_HAVE_X11
vpDisplayX d, d2, d3, d4, d5, d6;
#elif defined(VISP_HAVE_GDI)
vpDisplayGDI d, d2, d3, d4, d5, d6;
#elif defined(HAVE_OPENCV_HIGHGUI)
vpDisplayOpenCV d, d2, d3, d4, d5, d6;
#endif
d.init(I_color, 0, 0, "Input color image");
//! [Stretch contrast]
vpImage<vpRGBa> I_stretch;
vp::stretchContrast(I_color, I_stretch);
//! [Stretch contrast]
d2.init(I_stretch, I_color.getWidth(), 10, "Stretch contrast");
//! [Stretch contrast HSV]
vpImage<vpRGBa> I_stretch_hsv;
vp::stretchContrastHSV(I_color, I_stretch_hsv);
//! [Stretch contrast HSV]
d3.init(I_stretch_hsv, 0, I_color.getHeight() + 80, "Stretch contrast HSV");
//! [Histogram equalization]
vpImage<vpRGBa> I_hist_eq;
vp::equalizeHistogram(I_color, I_hist_eq);
//! [Histogram equalization]
d4.init(I_hist_eq, I_color.getWidth(), I_color.getHeight() + 80, "Histogram equalization");
//! [CLAHE]
vpImage<vpRGBa> I_clahe;
vp::clahe(I_color, I_clahe, blockRadius, bins, slope);
//! [CLAHE]
d5.init(I_clahe, 0, 2 * I_color.getHeight() + 80, "CLAHE");
//! [Unsharp mask]
vpImage<vpRGBa> I_unsharp;
vp::unsharpMask(I_clahe, I_unsharp, sigma, weight);
//! [Unsharp mask]
d6.init(I_unsharp, I_color.getWidth(), 2 * I_color.getHeight() + 80, "Unsharp mask");
vpDisplay::display(I_color);
vpDisplay::display(I_stretch);
vpDisplay::display(I_stretch_hsv);
vpDisplay::display(I_hist_eq);
vpDisplay::display(I_clahe);
vpDisplay::display(I_unsharp);
vpDisplay::displayText(I_unsharp, 20, 20, "Click to quit.", vpColor::red);
vpDisplay::flush(I_color);
vpDisplay::flush(I_stretch);
vpDisplay::flush(I_stretch_hsv);
vpDisplay::flush(I_hist_eq);
vpDisplay::flush(I_clahe);
vpDisplay::flush(I_unsharp);
vpDisplay::getClick(I_unsharp);
#else
(void)argc;
(void)argv;
#endif
return EXIT_SUCCESS;
}
|