From 85dd3fbe30e1bf9a194fd745ca6e7e7b55cbbaa2 Mon Sep 17 00:00:00 2001
From: Alexey Spizhevoy <no@email>
Date: Fri, 3 Dec 2010 12:31:20 +0000
Subject: [PATCH] fixed meanShiftSegmentation test in gpu module

---
 tests/gpu/src/mssegmentation.cpp | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/tests/gpu/src/mssegmentation.cpp b/tests/gpu/src/mssegmentation.cpp
index 77d8447a11..8a5bde43b2 100644
--- a/tests/gpu/src/mssegmentation.cpp
+++ b/tests/gpu/src/mssegmentation.cpp
@@ -64,10 +64,17 @@ struct CV_GpuMeanShiftSegmentationTest : public CvTest {
             Mat img;
             cvtColor(img_rgb, img, CV_BGR2BGRA);
 
-            for (int minsize = 0; minsize < 2000; minsize = (minsize + 1) * 2) 
+            int major, minor;
+            cv::gpu::getComputeCapability(cv::gpu::getDevice(), major, minor);
+
+            for (int minsize = 0; minsize < 2000; minsize = (minsize + 1) * 4) 
             {
                 stringstream path;
-                path << ts->get_data_path() << "meanshift/cones_segmented_sp10_sr10_minsize" << minsize << ".png";
+                path << ts->get_data_path() << "meanshift/cones_segmented_sp10_sr10_minsize" << minsize;
+                if (major == 1)
+                    path << "_CC1X.png";
+                else
+                    path << ".png";
 
                 Mat dst;
                 meanShiftSegmentation((GpuMat)img, dst, 10, 10, minsize);