WebSVN - OpenCV2-Cookbook - Diff - Rev 3 and 5 - /trunk/Chapter 04/imageComparator.h


/*------------------------------------------------------------------------------------------*\
   This file contains material supporting chapter 4 of the cookbook:  
   Computer Vision Programming using the OpenCV Library.
   by Robert Laganiere, Packt Publishing, 2011.

   This program is free software; permission is hereby granted to use, copy, modify,
   and distribute this source code, or portions thereof, for any purpose, without fee,
   subject to the restriction that the copyright notice may not be removed
   or altered from any source or altered source distribution.
   The software is released on an as-is basis and without any warranties of any kind.
   In particular, the software is not guaranteed to be fault-tolerant or free from failure.
   The author disclaims all warranties with regard to this software, any use,
   and any consequent failure, is purely the responsibility of the user.
 
   Copyright (C) 2010-2011 Robert Laganiere, www.laganiere.name
\*------------------------------------------------------------------------------------------*/

#if !defined ICOMPARATOR
#define ICOMPARATOR

#include <opencv2/core/core.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include "colorhistogram.h"

class ImageComparator {

  private:

        cv::Mat reference;
        cv::Mat input;
        cv::MatND refH;
        cv::MatND inputH;

        ColorHistogram hist;
        int div;

  public:

        ImageComparator() : div(32) {

        }

        // Color reduction factor
        // The comparaison will be made on images with
        // color space reduced by this factor in each dimension
        void setColorReduction( int factor) {

                div= factor;
        }

        int getColorReduction() {

                return div;
        }

        void setReferenceImage(const cv::Mat& image) {

                reference= hist.colorReduce(image,div);
                refH= hist.getHistogram(reference);
        }

        double compare(const cv::Mat& image) {

                input= hist.colorReduce(image,div);
                inputH= hist.getHistogram(input);

                return cv::compareHist(refH,inputH,CV_COMP_INTERSECT);
        }
};


#endif
 

Rev 3	Rev 5
1	/------------------------------------------------------------------------------------------\	1	/------------------------------------------------------------------------------------------\
2	This file contains material supporting chapter 4 of the cookbook:	2	This file contains material supporting chapter 4 of the cookbook:
3	Computer Vision Programming using the OpenCV Library.	3	Computer Vision Programming using the OpenCV Library.
4	by Robert Laganiere, Packt Publishing, 2011.	4	by Robert Laganiere, Packt Publishing, 2011.
5		5
6	This program is free software; permission is hereby granted to use, copy, modify,	6	This program is free software; permission is hereby granted to use, copy, modify,
7	and distribute this source code, or portions thereof, for any purpose, without fee,	7	and distribute this source code, or portions thereof, for any purpose, without fee,
8	subject to the restriction that the copyright notice may not be removed	8	subject to the restriction that the copyright notice may not be removed
9	or altered from any source or altered source distribution.	9	or altered from any source or altered source distribution.
10	The software is released on an as-is basis and without any warranties of any kind.	10	The software is released on an as-is basis and without any warranties of any kind.
11	In particular, the software is not guaranteed to be fault-tolerant or free from failure.	11	In particular, the software is not guaranteed to be fault-tolerant or free from failure.
12	The author disclaims all warranties with regard to this software, any use,	12	The author disclaims all warranties with regard to this software, any use,
13	and any consequent failure, is purely the responsibility of the user.	13	and any consequent failure, is purely the responsibility of the user.
14		14
15	Copyright (C) 2010-2011 Robert Laganiere, www.laganiere.name	15	Copyright (C) 2010-2011 Robert Laganiere, www.laganiere.name
16	\------------------------------------------------------------------------------------------/	16	\------------------------------------------------------------------------------------------/
17		17
18	#if !defined ICOMPARATOR	18	#if !defined ICOMPARATOR
19	#define ICOMPARATOR	19	#define ICOMPARATOR
20		20
21	#include <opencv2\core\core.hpp>	21	#include <opencv2/core/core.hpp>
22	#include <opencv2\imgproc\imgproc.hpp>	22	#include <opencv2/imgproc/imgproc.hpp>
23	#include "colorhistogram.h"	23	#include "colorhistogram.h"
24		24
25	class ImageComparator {	25	class ImageComparator {
26		26
27	private:	27	private:
28		28
29	cv::Mat reference;	29	cv::Mat reference;
30	cv::Mat input;	30	cv::Mat input;
31	cv::MatND refH;	31	cv::MatND refH;
32	cv::MatND inputH;	32	cv::MatND inputH;
33		33
34	ColorHistogram hist;	34	ColorHistogram hist;
35	int div;	35	int div;
36		36
37	public:	37	public:
38		38
39	ImageComparator() : div(32) {	39	ImageComparator() : div(32) {
40		40
41	}	41	}
42		42
43	// Color reduction factor	43	// Color reduction factor
44	// The comparaison will be made on images with	44	// The comparaison will be made on images with
45	// color space reduced by this factor in each dimension	45	// color space reduced by this factor in each dimension
46	void setColorReduction( int factor) {	46	void setColorReduction( int factor) {
47		47
48	div= factor;	48	div= factor;
49	}	49	}
50		50
51	int getColorReduction() {	51	int getColorReduction() {
52		52
53	return div;	53	return div;
54	}	54	}
55		55
56	void setReferenceImage(const cv::Mat& image) {	56	void setReferenceImage(const cv::Mat& image) {
57		57
58	reference= hist.colorReduce(image,div);	58	reference= hist.colorReduce(image,div);
59	refH= hist.getHistogram(reference);	59	refH= hist.getHistogram(reference);
60	}	60	}
61		61
62	double compare(const cv::Mat& image) {	62	double compare(const cv::Mat& image) {
63		63
64	input= hist.colorReduce(image,div);	64	input= hist.colorReduce(image,div);
65	inputH= hist.getHistogram(input);	65	inputH= hist.getHistogram(input);
66		66
67	return cv::compareHist(refH,inputH,CV_COMP_INTERSECT);	67	return cv::compareHist(refH,inputH,CV_COMP_INTERSECT);
68	}	68	}
69	};	69	};
70		70
71		71
72	#endif	72	#endif
73		73

Subversion Repositories OpenCV2-Cookbook

(root)/trunk/Chapter 04/imageComparator.h - Rev 3 → 5