Распознавание шаблонов OpenCV C++

У меня есть шаблон, и я хочу знать, присутствует ли шаблон в изображении. Ну, я много гуглил и пришел к выводу, что мне нужно использовать cvMatchTemplate а также cvMinMaxLoc,

Вот мой код:

image = cvLoadImage("C:/images/flower.jpg",1);
templat = cvLoadImage("C:/images/flo.jpg",1);
image2=cvCreateImage( cvSize(image->width, image->height), IPL_DEPTH_8U, 1 );
result=cvCreateImage( cvSize(image->width, image->height), IPL_DEPTH_8U, 1 );
cvZero(result);
cvZero(image2);
cvCvtColor(image,image2,CV_BGR2GRAY);
cvMatchTemplate(image2, templat,result,CV_TM_CCORR_NORMED);

double min_val=0, max_val=0;
CvPoint min_loc, max_loc;
cvMinMaxLoc(result, &min_val, &max_val, &min_loc, &max_loc);

cvRectangle(image, max_loc, cvPoint(max_loc.x+templat->width,  
max_loc.y+templat->height), cvScalar(0), 1);

cvShowImage( "src", image );
cvShowImage( "result image", result);
cvWaitKey(0);

Моя проблема заключается в том, что, когда я запускаю приведенный выше код, отображается окно с сообщением:

Unhandled exception at 0x747d812f in matching.exe: Microsoft C++ exception: cv::Exception at memory location 0x001ff6ec..

и на черном экране появляется сообщение:

OpenCV Error: Sizes of input arguments do not match <image and template should have the same type> in unknown function, file..\..\..\..\ocv\opencv\scr\cv\cvtempl.cpp, line 356.

Обратите внимание, что flower.jpg это цветное изображение и flo.jpg это серая шкала этого изображения.

Есть идеи о том, что происходит?

Источник

2 ответа

Решение

Вам нужно конвертировать оба flower.jpg а также flo.jpg на одноканальное изображение. Даже если flo.jpg в градациях серого, вы загружаете его как трехканальное изображение. Так же result изображение должно быть IPL_DEPTH_32F вместо IPL_DEPTH_8U,

Вот правильный код (не проверен):

IplImage* image    = cvLoadImage("C:/images/flower.jpg", 1);
IplImage* templat  = cvLoadImage("C:/images/flo.jpg", 1);
IplImage* image2   = cvCreateImage(cvSize(image->width, image->height), IPL_DEPTH_8U, 1);
IplImage* templat2 = cvCreateImage(cvSize(templat->width, templat->height), IPL_DEPTH_8U, 1); 

cvCvtColor(image, image2, CV_BGR2GRAY);
cvCvtColor(templat, templat2, CV_BGR2GRAY);

int w = image->width - templat->width + 1;
int h = image->height - templat->height + 1;
result = cvCreateImage(cvSize(w, h), IPL_DEPTH_32F, 1);
cvMatchTemplate(image2, templat, result, CV_TM_CCORR_NORMED);

double min_val, max_val;
CvPoint min_loc, max_loc;
cvMinMaxLoc(result, &min_val, &max_val, &min_loc, &max_loc);

cvRectangle(image, max_loc, cvPoint(max_loc.x+templat->width, 
max_loc.y+templat->height), cvScalar(0), 1);

cvShowImage("src", image);
cvShowImage("result image", result);
cvWaitKey(0);
Источник

Соответствие шаблона предполагает, что оба image а также template имеют одинаковое количество каналов и глубину канала. Самый простой способ сделать это - загрузить их в оттенках серого:

Mat I = imread("lena.png", 0);
Mat T = imread("template.png", 0);

Примечания: я бы приказал использовать интерфейс OpenCV2.0 C++. Так что вместо cvLoadImage использование imread, Старый интерфейс больше не разрабатывается.

Источник
Другие вопросы по тегам