首页 > 代码库 > 图像自动测试框架

图像自动测试框架

最近在做关于 Kinect 的项目,由于需要用到图像匹配相关的算法,从网上看到有大量的论文,一时之间也不知道哪些才是适合我的,于是乎我想到应该对这些方法进行试验,从而找到适合我的好方法。但是,问题来了,这么多的算法,一个一个的做实验,不现实啊。经过深思熟虑,我下定决心要写一个自动的框架,利用这个框架来加速我的试验速度。这个框架定义了三个基类:ImageGen、SkinDetect、Action。

ImageGen:自动的生成待检测的图像,接口如下:

class ImageGen{
public:
	virtual IplImage* operator()() = 0;
	virtual IplImage* getImage() = 0;
	virtual void	  reset() = 0;
};

operator() 函数与 getImage() 功能应该是一致的,主要就是获得图像,为了方便与其他的类进行交互,这里生成的图像格式应该是:BGR 的,然后其它的检测方法如果利用的是其它的图片格式,则可以做适当的转换,reset() 重置指针,重新获取图像。

这所以定义这个类是因为想到扩展的问题,这样无论图像来自何方,只要符合该接口标准就可以使用该框架了。


SkinDetect:图像检测类,这个类主要完成的功能是检测图像,由于我的试验是有关肤色检测的,所以这个类就跟肤色检测相关了,如果为了扩展,可以进一步抽象,成为更基本的类:ImageDetect...,接口如下:

class SkinDetect{
public:
	SkinDetect()
		: _skin_counts(0)
		, _goods_counts(0)
	{}

	// src 与 res 的 row 和 col 一致, src 的格式一律为 BGR
	// 结果存放在 res 中
	virtual bool detected( IplImage* src, IplImage* res ) = 0;
	// 参数字符串化,这样有助于判断当前的参数值。
	virtual std::string toString() = 0;
	virtual ~SkinDetect() {}

	size_t getSkinCounts() { return _skin_counts;  }
	size_t getGoodsCounts(){ return _goods_counts; }

	// 结果图像中的灰度值含义:
	static const int BG_VALUE    = http://www.mamicode.com/0;		// 背景灰度值>
之所以定义了 goods_counts 主要是因为这次实验跟商品也有关系,因此就把它放进基类了。值得注意的是,还定义了静态常量:BG_VALUE、SKIN_VALUE、GOODS_VALUE,这是因为我把图像分成了三部分:皮肤、商品、背景。不同的部分对应不同的颜色。

Action:动作类,这个类的功能主要是当检测返回 true 的时候执行相应的动作,接口如下:

/***********************************************
* 当判断符合条件的时候调用该类的方法进行处理
***********************************************/
class Action{
public:
	virtual void operator()(IplImage* src, IplImage* res) = 0;
	virtual ~Action(){};
};
只有一个公共接口:operator(),第一个参数为原图像,第二个参数为检测结果。


最后一个类是:JudgerFrameWork,该类引用上面的类进行自动化的检测,如下:

class JudgerFrameWork{
public:
	JudgerFrameWork( ImageGen& img_gen, SkinDetect& skin_detect, Action& action )
		: _img_gen(img_gen)
		, _skin_detect(skin_detect)
		, _action(action)
	{}
	void start()
	{
		for( IplImage* image = _img_gen.getImage(); image != NULL; image = _img_gen.getImage() )
		{
			IplImage* res = cvCreateImage( cvGetSize(image), IPL_DEPTH_8U, 1 );
			if( _skin_detect.detected( image, res ) )
			{
				_action( image, res );
				std::cout << "goods count:" << _skin_detect.getGoodsCounts() << std::endl;
				std::cout << "skins count:" << _skin_detect.getSkinCounts()  << std::endl;
			}
		}
	}
private:
	ImageGen& _img_gen;
	SkinDetect& _skin_detect;
	Action& _action;
};
所需要做的事情主要是创建相应的对象,然后构造 JudgerFrameWork 对象,当构造完对象后,只需调用函数:start() 即可完成所有的动作了。


设置这套框架前前后后修改了 N 次,主要是因为接口那里难以控制,有时候很难协调功能的分配。此外,如果设置的太过抽象,则又增加了开发的复杂度,如果抽象度不够,则又不利于扩展。

不禁感叹:开发大有学问在,接口设计功夫深!

图像自动测试框架