基于边缘检测和HSV的图像识别算法-main.cpp

#include "DryDetect.h"

int main()
{
	Mat src = cv::imread("H:\\ImageProcess\\Dry\\image\\7.jpg");
	/*
	-1:传入图片错误
	0:没有检测到框
	1:成功检测到了框
	*/
	DryParam DP;
	//以下参数做成配置文件可修改
	DP.minWidth = 80;   // 可调参数，限制格子的最大最小宽高
	DP.minHeight = 80;
	DP.maxWidth = 180;
	DP.maxHeight = 180;
	DP.drydetect = 1; //if 0, 不用算法定位

	std::vector<cv::Rect> resultBoxes; //算法识别结果
	/*
	Dryflag:
	-1: 传入图片错误
	0:  没有检测到框
	1:  成功检测到了框
	*/
	int Dryflag;
	if (DP.drydetect = 0)  //0: 不调用算法定位
	{
		Dryflag = 0;
	}
	else
	{
		Dryflag = DryAlg(src, resultBoxes, DP);
	}

	switch (Dryflag)
	{
	case -1:
	{
		cout << "load image error..." << endl;
		break;
	}
	case 0:
	{
		cout << "detect box fail..." << endl;
		break;
	}
	default:
		break;
	}
	return 0;
}