Home | History | Annotate | Download | only in Histograms_Matching
      1 /**
      2  * @file compareHist_Demo.cpp
      3  * @brief Sample code to use the function compareHist
      4  * @author OpenCV team
      5  */
      6 
      7 #include "opencv2/imgcodecs.hpp"
      8 #include "opencv2/highgui/highgui.hpp"
      9 #include "opencv2/imgproc/imgproc.hpp"
     10 #include <iostream>
     11 #include <stdio.h>
     12 
     13 using namespace std;
     14 using namespace cv;
     15 
     16 /**
     17  * @function main
     18  */
     19 int main( int argc, char** argv )
     20 {
     21     Mat src_base, hsv_base;
     22     Mat src_test1, hsv_test1;
     23     Mat src_test2, hsv_test2;
     24     Mat hsv_half_down;
     25 
     26     /// Load three images with different environment settings
     27     if( argc < 4 )
     28     {
     29         printf("** Error. Usage: ./compareHist_Demo <image_settings0> <image_setting1> <image_settings2>\n");
     30         return -1;
     31     }
     32 
     33     src_base = imread( argv[1], 1 );
     34     src_test1 = imread( argv[2], 1 );
     35     src_test2 = imread( argv[3], 1 );
     36 
     37     /// Convert to HSV
     38     cvtColor( src_base, hsv_base, COLOR_BGR2HSV );
     39     cvtColor( src_test1, hsv_test1, COLOR_BGR2HSV );
     40     cvtColor( src_test2, hsv_test2, COLOR_BGR2HSV );
     41 
     42     hsv_half_down = hsv_base( Range( hsv_base.rows/2, hsv_base.rows - 1 ), Range( 0, hsv_base.cols - 1 ) );
     43 
     44     /// Using 50 bins for hue and 60 for saturation
     45     int h_bins = 50; int s_bins = 60;
     46     int histSize[] = { h_bins, s_bins };
     47 
     48     // hue varies from 0 to 179, saturation from 0 to 255
     49     float h_ranges[] = { 0, 180 };
     50     float s_ranges[] = { 0, 256 };
     51 
     52     const float* ranges[] = { h_ranges, s_ranges };
     53 
     54     // Use the o-th and 1-st channels
     55     int channels[] = { 0, 1 };
     56 
     57 
     58     /// Histograms
     59     MatND hist_base;
     60     MatND hist_half_down;
     61     MatND hist_test1;
     62     MatND hist_test2;
     63 
     64     /// Calculate the histograms for the HSV images
     65     calcHist( &hsv_base, 1, channels, Mat(), hist_base, 2, histSize, ranges, true, false );
     66     normalize( hist_base, hist_base, 0, 1, NORM_MINMAX, -1, Mat() );
     67 
     68     calcHist( &hsv_half_down, 1, channels, Mat(), hist_half_down, 2, histSize, ranges, true, false );
     69     normalize( hist_half_down, hist_half_down, 0, 1, NORM_MINMAX, -1, Mat() );
     70 
     71     calcHist( &hsv_test1, 1, channels, Mat(), hist_test1, 2, histSize, ranges, true, false );
     72     normalize( hist_test1, hist_test1, 0, 1, NORM_MINMAX, -1, Mat() );
     73 
     74     calcHist( &hsv_test2, 1, channels, Mat(), hist_test2, 2, histSize, ranges, true, false );
     75     normalize( hist_test2, hist_test2, 0, 1, NORM_MINMAX, -1, Mat() );
     76 
     77     /// Apply the histogram comparison methods
     78     for( int i = 0; i < 4; i++ )
     79     {
     80         int compare_method = i;
     81         double base_base = compareHist( hist_base, hist_base, compare_method );
     82         double base_half = compareHist( hist_base, hist_half_down, compare_method );
     83         double base_test1 = compareHist( hist_base, hist_test1, compare_method );
     84         double base_test2 = compareHist( hist_base, hist_test2, compare_method );
     85 
     86         printf( " Method [%d] Perfect, Base-Half, Base-Test(1), Base-Test(2) : %f, %f, %f, %f \n", i, base_base, base_half , base_test1, base_test2 );
     87     }
     88 
     89     printf( "Done \n" );
     90 
     91     return 0;
     92 }
     93