OpenCV2计算机视觉应用编程手册（自学版）初级六

OpenCV2计算机视觉应用编程手册（自学版）初级六

第六个opencv程序---类的学习
天空区域识别、图像分割

如果没接触过类最好稍微读一下C++中类的介绍，不用很深，一点知识就够了。整个封装做到比较完美(对于我这个入门的来说)。

本部分主要包含三个文件，sys_main.c文件，detector.cpp文件，detector.h文件sys_main.c文件就是我们的主文件，detector.h文件是我们声明的类的存放文件，detector.cpp文件，是声明的类的虚函数定义文件。

1---------sys_main.c文件

用类声明一个对象，然后初始化对象，并调用对象的函数进行处理。

/*

第六个opencv程序---类的学习，识别天空区域，图像分割

时间：2014年12月1日22:17:53

整理：天空之恋

测试类的使用，使用C++中的类处理

*/

#include "stdafx.h"

#include<iostream>

#include "detector.h"

#include <opencv2/opencv.hpp>

using namespace std;// 使用STD

using namespace cv;// 使用名字空间

int main()

{

// Create image processor object

ColorDetector cdetect;// 声明图像处理的类对象



// Read input image

cv::Mat image= cv::imread("F:\\house.jpg");

if (!image.data)

return 0;



// set input parameters

cdetect.setTargetColor(130,190,230); // here blue sky这个像素代表的是天空区域的值，这样我们把每个点

//的像素与这个点进行比较，最后我们就可以

//识别出天空区域



// Read image, process it and display the result

cv::namedWindow("result");

cv::imshow("result",cdetect.process(image));//调用类封装中的处理函数process



cv::waitKey();



return 0;

}

2--------detector.cpp文件

对声明的类的虚函数process进行重新定义，实现计算距离的功能。

#include "detector.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;

}

3---------detector.h文件

类的定义，私有部分，公有部分，处理函数等。

#if !defined COLORDETECT

#define COLORDETECT



#include <opencv2/opencv.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 设置需要比较的像素的值// 这个地方需要注意了BGR B->target[0]

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