added gpu RGB<->Lab and BGR<->Luv conversions
This commit is contained in:
@@ -43,8 +43,6 @@
|
||||
|
||||
#ifdef HAVE_CUDA
|
||||
|
||||
#include <cuda_runtime_api.h>
|
||||
|
||||
using namespace std;
|
||||
using namespace cv;
|
||||
using namespace cv::gpu;
|
||||
|
||||
@@ -72,4 +72,9 @@
|
||||
#include "utility.hpp"
|
||||
#include "interpolation.hpp"
|
||||
|
||||
#ifdef HAVE_CUDA
|
||||
#include <cuda.h>
|
||||
#include <cuda_runtime.h>
|
||||
#endif
|
||||
|
||||
#endif
|
||||
|
||||
@@ -1628,7 +1628,38 @@ TEST_P(CvtColor, BGR2Lab)
|
||||
}
|
||||
catch (const cv::Exception& e)
|
||||
{
|
||||
#if (CUDA_VERSION < 5000)
|
||||
ASSERT_EQ(CV_StsBadFlag, e.code);
|
||||
#else
|
||||
FAIL();
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
TEST_P(CvtColor, RGB2Lab)
|
||||
{
|
||||
if (depth != CV_8U)
|
||||
return;
|
||||
|
||||
try
|
||||
{
|
||||
cv::Mat src = readImage("stereobm/aloe-L.png");
|
||||
|
||||
cv::gpu::GpuMat dst_lab = createMat(src.size(), src.type(), useRoi);
|
||||
cv::gpu::cvtColor(loadMat(src, useRoi), dst_lab, cv::COLOR_RGB2Lab);
|
||||
|
||||
cv::gpu::GpuMat dst_bgr = createMat(src.size(), src.type(), useRoi);
|
||||
cv::gpu::cvtColor(dst_lab, dst_bgr, cv::COLOR_Lab2RGB);
|
||||
|
||||
EXPECT_MAT_NEAR(src, dst_bgr, 10);
|
||||
}
|
||||
catch (const cv::Exception& e)
|
||||
{
|
||||
#if (CUDA_VERSION < 5000)
|
||||
ASSERT_EQ(CV_StsBadFlag, e.code);
|
||||
#else
|
||||
FAIL();
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1641,6 +1672,33 @@ TEST_P(CvtColor, BGR2Luv)
|
||||
{
|
||||
cv::Mat src = img;
|
||||
|
||||
cv::gpu::GpuMat dst_luv = createMat(src.size(), src.type(), useRoi);
|
||||
cv::gpu::cvtColor(loadMat(src, useRoi), dst_luv, cv::COLOR_BGR2Luv);
|
||||
|
||||
cv::gpu::GpuMat dst_rgb = createMat(src.size(), src.type(), useRoi);
|
||||
cv::gpu::cvtColor(dst_luv, dst_rgb, cv::COLOR_Luv2BGR);
|
||||
|
||||
EXPECT_MAT_NEAR(src, dst_rgb, 10);
|
||||
}
|
||||
catch (const cv::Exception& e)
|
||||
{
|
||||
#if (CUDA_VERSION < 5000)
|
||||
ASSERT_EQ(CV_StsBadFlag, e.code);
|
||||
#else
|
||||
FAIL();
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
TEST_P(CvtColor, RGB2Luv)
|
||||
{
|
||||
if (depth != CV_8U)
|
||||
return;
|
||||
|
||||
try
|
||||
{
|
||||
cv::Mat src = img;
|
||||
|
||||
cv::gpu::GpuMat dst_luv = createMat(src.size(), src.type(), useRoi);
|
||||
cv::gpu::cvtColor(loadMat(src, useRoi), dst_luv, cv::COLOR_RGB2Luv);
|
||||
|
||||
@@ -1651,7 +1709,11 @@ TEST_P(CvtColor, BGR2Luv)
|
||||
}
|
||||
catch (const cv::Exception& e)
|
||||
{
|
||||
#if (CUDA_VERSION < 5000)
|
||||
ASSERT_EQ(CV_StsBadFlag, e.code);
|
||||
#else
|
||||
FAIL();
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user