fixed meanShift test in gpu module
authorAlexey Spizhevoy <no@email>
Fri, 3 Dec 2010 12:13:19 +0000 (12:13 +0000)
committerAlexey Spizhevoy <no@email>
Fri, 3 Dec 2010 12:13:19 +0000 (12:13 +0000)
tests/gpu/src/meanshift.cpp

index 42fc2ce..1c32550 100644 (file)
@@ -54,7 +54,15 @@ struct CV_GpuMeanShiftTest : public CvTest
         int colorRad = 30;\r
 \r
         cv::Mat img = cv::imread(std::string(ts->get_data_path()) + "meanshift/cones.png");\r
-        cv::Mat img_template = cv::imread(std::string(ts->get_data_path()) + "meanshift/con_result.png");\r
+        cv::Mat img_template;\r
+        \r
+        int major, minor;\r
+        cv::gpu::getComputeCapability(cv::gpu::getDevice(), major, minor);\r
+        \r
+        if (major == 1)\r
+            img_template = cv::imread(std::string(ts->get_data_path()) + "meanshift/con_result_CC1X.png");\r
+        else\r
+            img_template = cv::imread(std::string(ts->get_data_path()) + "meanshift/con_result.png");\r
 \r
         if (img.empty() || img_template.empty())\r
         {\r
@@ -69,6 +77,7 @@ struct CV_GpuMeanShiftTest : public CvTest
         {\r
             cv::gpu::GpuMat res;\r
             cv::gpu::meanShiftFiltering( cv::gpu::GpuMat(rgba), res, spatialRad, colorRad );\r
+\r
             if (res.type() != CV_8UC4)\r
             {\r
                 ts->set_failed_test_info(CvTS::FAIL_INVALID_OUTPUT);\r
@@ -197,7 +206,7 @@ struct CV_GpuMeanShiftProcTest : public CvTest
             cv::gpu::getComputeCapability(cv::gpu::getDevice(), major, minor);\r
 \r
             if (major == 1)\r
-                fs.open(std::string(ts->get_data_path()) + "meanshift/spmap_1x.yaml", cv::FileStorage::READ);\r
+                fs.open(std::string(ts->get_data_path()) + "meanshift/spmap_CC1X.yaml", cv::FileStorage::READ);\r
             else\r
                 fs.open(std::string(ts->get_data_path()) + "meanshift/spmap.yaml", cv::FileStorage::READ);\r
             fs["spmap"] >> spmap_template;\r