generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

refactored test code

Browse Source
This commit is contained in:
marina.kolpakova 2012-08-08 14:15:36 +04:00
parent c5e92c31ba
commit 490622231a
1 changed files with 19 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
modules/gpu/test/test_labeling.cpp
View File

@ -74,7 +74,7 @@ namespace {
}; };
GreedyLabeling(cv::Mat img) GreedyLabeling(cv::Mat img)
: image(img), _labels(image.cols, image.rows, CV_32SC1, cv::Scalar::all(-1)) {} : image(img), _labels(image.size(), CV_32SC1, cv::Scalar::all(-1)) {}
void operator() (cv::Mat labels) const void operator() (cv::Mat labels) const
{ {
@ -106,33 +106,23 @@ namespace {
int* dl = &dist_labels[p.y * pitch + p.x]; int* dl = &dist_labels[p.y * pitch + p.x];
unsigned char* sp = &source[p.y * image.step1() + p.x]; unsigned char* sp = &source[p.y * image.step1() + p.x];
dl[0] = cc;
//right //right
if( p.x < (width - 1) && dl[ +1] == -1 && inInt(sp[0], sp[+1])) if( p.x < (width - 1) && dl[ +1] == -1 && inInt(sp[0], sp[+1]))
{
dl[+1] = cc;
*top++ = dot::make(p.x + 1, p.y); *top++ = dot::make(p.x + 1, p.y);
}
//left //left
if( p.x > 0 && dl[-1] == -1 && inInt(sp[0], sp[-1])) if( p.x > 0 && dl[-1] == -1 && inInt(sp[0], sp[-1]))
{
dl[-1] = cc;
*top++ = dot::make(p.x - 1, p.y); *top++ = dot::make(p.x - 1, p.y);
}
//bottom //bottom
if( p.y < (height - 1) && dl[+pitch] == -1 && inInt(sp[0], sp[+pitch])) if( p.y < (height - 1) && dl[+pitch] == -1 && inInt(sp[0], sp[+image.step1()]))
{
dl[+pitch] = cc;
*top++ = dot::make(p.x, p.y + 1); *top++ = dot::make(p.x, p.y + 1);
}
//top //top
if( p.y > 0 && dl[-pitch] == -1 && inInt(sp[0], sp[-pitch])) if( p.y > 0 && dl[-pitch] == -1 && inInt(sp[0], sp[-image.step1()]))
{
dl[-pitch] = cc;
*top++ = dot::make(p.x, p.y - 1); *top++ = dot::make(p.x, p.y - 1);
}
p = *--top; p = *--top;
} }
@ -171,6 +161,10 @@ TEST_P(Labeling, ConnectedComponents)
GreedyLabeling host(image); GreedyLabeling host(image);
host(host._labels); host(host._labels);
double minVal, maxVal;
cv::minMaxLoc(host._labels, &minVal, &maxVal);
std::cout << minVal << " " << maxVal << std::endl;
cv::gpu::GpuMat mask; cv::gpu::GpuMat mask;
mask.create(image.rows, image.cols, CV_8UC1); mask.create(image.rows, image.cols, CV_8UC1);
@ -181,18 +175,20 @@ TEST_P(Labeling, ConnectedComponents)
ASSERT_NO_THROW(cv::gpu::labelComponents(mask, components)); ASSERT_NO_THROW(cv::gpu::labelComponents(mask, components));
// for (int j = 0; j + 32 < components.rows; j += 32) for (int j = 0; j + 32 < components.rows; j += 32)
// for (int i = 0; i + 32 < components.cols; i += 32) for (int i = 0; i + 32 < components.cols; i += 32)
// { {
// std::cout << "Tile: " << i << " " << j << std::endl; std::cout << "Tile: " << i << " " << j << std::endl;
// std::cout << cv::Mat(host._labels, cv::Rect(i,j,32,32)) << std::endl;
std::cout << cv::Mat(image, cv::Rect(i,j,32,32)) << std::endl;
std::cout << cv::Mat(host._labels, cv::Rect(i,j,32,32)) << std::endl;
// std::cout << cv::Mat(cv::Mat(components), cv::Rect(i,j,32,32)) << std::endl; // std::cout << cv::Mat(cv::Mat(components), cv::Rect(i,j,32,32)) << std::endl;
// } }
// for debug // for debug
// cv::imshow("test", image); // cv::imshow("test", image);
// cv::waitKey(0); // cv::waitKey(0);
// cv::imshow("test", host._labels * 50); // cv::imshow("test", host._labels * 5);
// cv::waitKey(0); // cv::waitKey(0);
// // cv::imshow("test", cv::Mat(mask) * 10); // // cv::imshow("test", cv::Mat(mask) * 10);
// // cv::waitKey(0); // // cv::waitKey(0);