From fbc1ac7dc402162fadfa89a05e2916e3fd5b6706 Mon Sep 17 00:00:00 2001 From: alexeylatyshev Date: Tue, 26 Jan 2010 14:04:30 +0000 Subject: [PATCH] Mser regression test was added git-svn-id: https://code.ros.org/svn/opencv/trunk@2626 73c94f0f-984f-4a5f-82bc-2d8db8d8ee08 --- opencv/tests/cv/src/amser.cpp | 201 ++++++++++++++++++++++++++++++++++ 1 file changed, 201 insertions(+) create mode 100644 opencv/tests/cv/src/amser.cpp diff --git a/opencv/tests/cv/src/amser.cpp b/opencv/tests/cv/src/amser.cpp new file mode 100644 index 00000000..4c3085d8 --- /dev/null +++ b/opencv/tests/cv/src/amser.cpp @@ -0,0 +1,201 @@ +/*M/////////////////////////////////////////////////////////////////////////////////////// +// +// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING. +// +// By downloading, copying, installing or using the software you agree to this license. +// If you do not agree to this license, do not download, install, +// copy or use the software. +// +// +// License Agreement +// For Open Source Computer Vision Library +// +// Copyright (C) 2000-2008, Intel Corporation, all rights reserved. +// Copyright (C) 2009, Willow Garage Inc., all rights reserved. +// Third party copyrights are property of their respective owners. +// +// Redistribution and use in source and binary forms, with or without modification, +// are permitted provided that the following conditions are met: +// +// * Redistribution's of source code must retain the above copyright notice, +// this list of conditions and the following disclaimer. +// +// * Redistribution's in binary form must reproduce the above copyright notice, +// this list of conditions and the following disclaimer in the documentation +// and/or other materials provided with the distribution. +// +// * The name of the copyright holders may not be used to endorse or promote products +// derived from this software without specific prior written permission. +// +// This software is provided by the copyright holders and contributors "as is" and +// any express or implied warranties, including, but not limited to, the implied +// warranties of merchantability and fitness for a particular purpose are disclaimed. +// In no event shall the Intel Corporation or contributors be liable for any direct, +// indirect, incidental, special, exemplary, or consequential damages +// (including, but not limited to, procurement of substitute goods or services; +// loss of use, data, or profits; or business interruption) however caused +// and on any theory of liability, whether in contract, strict liability, +// or tort (including negligence or otherwise) arising in any way out of +// the use of this software, even if advised of the possibility of such damage. +// +//M*/ + +#include "cvtest.h" +#include +#include +using namespace std; +using namespace cv; + +class CV_MserTest : public CvTest +{ +public: + CV_MserTest(); +protected: + void run(int); + int LoadBoxes(const char* path, vector& boxes); + int SaveBoxes(const char* path, const vector& boxes); + int CompareBoxes(const vector& boxes1,const vector& boxes2, float max_rel_diff = 0.01f); +}; + +CV_MserTest::CV_MserTest() : CvTest("MSER","cvExtractMSER") +{ +} + +int CV_MserTest::LoadBoxes(const char* path, vector& boxes) +{ + boxes.clear(); + FILE* f = fopen(path,"r"); + + if (f==NULL) + { + return 0; + } + + while (!feof(f)) + { + CvBox2D box; + fscanf(f,"%f,%f,%f,%f,%f\n",&box.angle,&box.center.x,&box.center.y,&box.size.width,&box.size.height); + boxes.push_back(box); + } + fclose(f); + return 1; +} + +int CV_MserTest::SaveBoxes(const char* path, const vector& boxes) +{ + FILE* f = fopen(path,"w"); + if (f==NULL) + { + return 0; + } + for (int i=0;i<(int)boxes.size();i++) + { + fprintf(f,"%f,%f,%f,%f,%f\n",boxes[i].angle,boxes[i].center.x,boxes[i].center.y,boxes[i].size.width,boxes[i].size.height); + } + fclose(f); + return 1; +} + +int CV_MserTest::CompareBoxes(const vector& boxes1,const vector& boxes2, float max_rel_diff) +{ + if (boxes1.size() != boxes2.size()) + return 0; + + for (int i=0; i<(int)boxes1.size();i++) + { + float rel_diff; + if (!((boxes1[i].angle == 0.0f) && (abs(boxes2[i].angle) < max_rel_diff))) + { + rel_diff = abs(boxes1[i].angle-boxes2[i].angle)/abs(boxes1[i].angle); + if (rel_diff > max_rel_diff) + return i; + } + + if (!((boxes1[i].center.x == 0.0f) && (abs(boxes2[i].center.x) < max_rel_diff))) + { + rel_diff = abs(boxes1[i].center.x-boxes2[i].center.x)/abs(boxes1[i].center.x); + if (rel_diff > max_rel_diff) + return i; + } + + if (!((boxes1[i].center.y == 0.0f) && (abs(boxes2[i].center.y) < max_rel_diff))) + { + rel_diff = abs(boxes1[i].center.y-boxes2[i].center.y)/abs(boxes1[i].center.y); + if (rel_diff > max_rel_diff) + return i; + } + if (!((boxes1[i].size.width == 0.0f) && (abs(boxes2[i].size.width) < max_rel_diff))) + { + rel_diff = abs(boxes1[i].size.width-boxes2[i].size.width)/abs(boxes1[i].size.width); + if (rel_diff > max_rel_diff) + return i; + } + + if (!((boxes1[i].size.height == 0.0f) && (abs(boxes2[i].size.height) < max_rel_diff))) + { + rel_diff = abs(boxes1[i].size.height-boxes2[i].size.height)/abs(boxes1[i].size.height); + if (rel_diff > max_rel_diff) + return i; + } + } + + return -1; +} + +void CV_MserTest::run(int) +{ + string image_path = string(ts->get_data_path()) + "mser/puzzle.png"; + + IplImage* img = cvLoadImage( image_path.c_str()); + if (!img) + { + ts->printf( CvTS::LOG, "Unable to open image mser/puzzle.png\n"); + ts->set_failed_test_info(CvTS::FAIL_MISSING_TEST_DATA); + return; + } + + CvSeq* contours; + CvMemStorage* storage= cvCreateMemStorage(); + IplImage* hsv = cvCreateImage( cvGetSize( img ), IPL_DEPTH_8U, 3 ); + cvCvtColor( img, hsv, CV_BGR2YCrCb ); + CvMSERParams params = cvMSERParams();//cvMSERParams( 5, 60, cvRound(.2*img->width*img->height), .25, .2 ); + cvExtractMSER( hsv, NULL, &contours, storage, params ); + + vector boxes; + vector boxes_orig; + for ( int i = 0; i < contours->total; i++ ) + { + CvContour* r = *(CvContour**)cvGetSeqElem( contours, i ); + CvBox2D box = cvFitEllipse2( r ); + box.angle=(float)CV_PI/2-box.angle; + boxes.push_back(box); + } + + string boxes_path = string(ts->get_data_path()) + "mser/boxes.txt"; + + if (!LoadBoxes(boxes_path.c_str(),boxes_orig)) + { + SaveBoxes(boxes_path.c_str(),boxes); + ts->printf( CvTS::LOG, "Unable to open data file mser/boxes.txt\n"); + ts->set_failed_test_info(CvTS::FAIL_MISSING_TEST_DATA); + return; + } + + const float dissimularity = 0.01f; + int n_box = CompareBoxes(boxes_orig,boxes,dissimularity); + if (n_box < 0) + { + ts->set_failed_test_info(CvTS::OK); + } + else + { + ts->set_failed_test_info(CvTS::FAIL_BAD_ACCURACY); + ts->printf( CvTS::LOG, "Incorrect correspondence in %d box\n",n_box); + } + + cvReleaseMemStorage(&storage); + cvReleaseImage(&hsv); + cvReleaseImage(&img); +} + +CV_MserTest mser_test; \ No newline at end of file -- 2.39.2