一起学opencv (五) 图片的数据计较，类的应用

简单的比较彩色图数据RGB值分别于设定的三个色度值比较，大则输出白色，小则输出黑色值。

colordetector.h

#if !defined COLORDETECT
#define COLORDETECT

#include <opencv2/core/core.hpp>

class ColorDetector {

private:

	// minimum acceptable distance
	int minDist;

	// target color
	cv::Vec3b target;

	// image containing resulting binary map
	cv::Mat result;

	// inline private member function
	// Computes the distance from target color.
	int getDistance(const cv::Vec3b& color) const {
		// return static_cast<int>(cv::norm<int,3>(cv::Vec3i(color[0]-target[0],color[1]-target[1],color[2]-target[2])));
		return abs(color[0] - target[0]) +
			abs(color[1] - target[1]) +
			abs(color[2] - target[2]);
	}

public:

	// empty constructor
	ColorDetector() : minDist(100) {

		// default parameter initialization here
		target[0] = target[1] = target[2] = 0;
	}

	// Getters and setters

	// Sets the color distance threshold.
	// Threshold must be positive, otherwise distance threshold
	// is set to 0.
	void setColorDistanceThreshold(int distance) {

		if (distance<0)
			distance = 0;
		minDist = distance;
	}

	// Gets the color distance threshold
	int getColorDistanceThreshold() const {

		return minDist;
	}

	// Sets the color to be detected
	void setTargetColor(unsigned char red, unsigned char green, unsigned char blue) {

		target[2] = red;
		target[1] = green;
		target[0] = blue;
	}

	// Sets the color to be detected
	void setTargetColor(cv::Vec3b color) {

		target = color;
	}

	// Gets the color to be detected
	cv::Vec3b getTargetColor() const {

		return target;
	}

	// Processes the image. Returns a 1-channel binary image.
	cv::Mat process(const cv::Mat &image);
};


#endif

 

类函数cv::Mat process(const cv::Mat &image)在头文件

colordetector.cpp

#include "colordetector.h"

cv::Mat ColorDetector::process(const cv::Mat &image) {

	// re-allocate binary map if necessary
	// same size as input image, but 1-channel
	result.create(image.rows, image.cols, CV_8U);

	// get the iterators
	cv::Mat_<cv::Vec3b>::const_iterator it = image.begin<cv::Vec3b>();
	cv::Mat_<cv::Vec3b>::const_iterator itend = image.end<cv::Vec3b>();
	cv::Mat_<uchar>::iterator itout = result.begin<uchar>();

	// for each pixel
	for (; it != itend; ++it, ++itout) {

		// process each pixel ---------------------

		// compute distance from target color
		if (getDistance(*it)<minDist) {

			*itout = 255;

		}
		else {

			*itout = 0;
		}

		// end of pixel processing ----------------
	}

	return result;
}

主程序

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>

#include "colordetector.h"

int main()
{
	// Create image processor object
	ColorDetector cdetect;

	// Read input image
	cv::Mat image = cv::imread("bfg.jpg");
	if (!image.data)
		return 0;

	// set input parameters
	cdetect.setTargetColor(130, 130, 90); // here blue sky

	// Read image, process it and display the result
	cv::Mat image2 = cdetect.process(image);
	cv::namedWindow("result");
	cv::imshow("result",image2);
	cv::imwrite("saved.jpg", image2);
	cv::waitKey();

	return 0;
}

结果