mirror of
https://github.com/opencv/opencv.git
synced 2024-11-27 04:36:36 +08:00
128 lines
4.0 KiB
C++
128 lines
4.0 KiB
C++
// This file is part of OpenCV project.
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
|
// of this distribution and at http://opencv.org/license.html.
|
|
|
|
// Copyright (C) 2014, Advanced Micro Devices, Inc., all rights reserved.
|
|
// Third party copyrights are property of their respective owners.
|
|
|
|
#include "../test_precomp.hpp"
|
|
#include "opencv2/ts/ocl_test.hpp"
|
|
|
|
#ifdef HAVE_OPENCL
|
|
|
|
namespace cvtest {
|
|
namespace ocl {
|
|
|
|
PARAM_TEST_CASE(FastNlMeansDenoisingTestBase, Channels, bool, bool)
|
|
{
|
|
int cn, templateWindowSize, searchWindowSize;
|
|
float h;
|
|
bool use_roi, use_image;
|
|
|
|
TEST_DECLARE_INPUT_PARAMETER(src);
|
|
TEST_DECLARE_OUTPUT_PARAMETER(dst);
|
|
|
|
virtual void SetUp()
|
|
{
|
|
cn = GET_PARAM(0);
|
|
use_roi = GET_PARAM(1);
|
|
use_image = GET_PARAM(2);
|
|
|
|
templateWindowSize = 7;
|
|
searchWindowSize = 21;
|
|
h = 3.0f;
|
|
}
|
|
|
|
virtual void generateTestData()
|
|
{
|
|
const int type = CV_8UC(cn);
|
|
Mat image;
|
|
|
|
if (use_image) {
|
|
image = readImage("denoising/lena_noised_gaussian_sigma=10.png",
|
|
cn == 1 ? IMREAD_GRAYSCALE : IMREAD_COLOR);
|
|
ASSERT_FALSE(image.empty());
|
|
}
|
|
|
|
Size roiSize = use_image ? image.size() : randomSize(1, MAX_VALUE);
|
|
Border srcBorder = randomBorder(0, use_roi ? MAX_VALUE : 0);
|
|
randomSubMat(src, src_roi, roiSize, srcBorder, type, 0, 255);
|
|
if (use_image) {
|
|
ASSERT_TRUE(cn == 1 || cn == 2 || cn == 3 || cn == 4);
|
|
if (cn == 2) {
|
|
int from_to[] = { 0,0, 1,1 };
|
|
src_roi.create(roiSize, type);
|
|
mixChannels(&image, 1, &src_roi, 1, from_to, 2);
|
|
}
|
|
else if (cn == 4) {
|
|
int from_to[] = { 0,0, 1,1, 2,2, 1,3};
|
|
src_roi.create(roiSize, type);
|
|
mixChannels(&image, 1, &src_roi, 1, from_to, 4);
|
|
}
|
|
else image.copyTo(src_roi);
|
|
}
|
|
|
|
Border dstBorder = randomBorder(0, use_roi ? MAX_VALUE : 0);
|
|
randomSubMat(dst, dst_roi, roiSize, dstBorder, type, 0, 255);
|
|
|
|
UMAT_UPLOAD_INPUT_PARAMETER(src);
|
|
UMAT_UPLOAD_OUTPUT_PARAMETER(dst);
|
|
}
|
|
};
|
|
|
|
typedef FastNlMeansDenoisingTestBase FastNlMeansDenoising;
|
|
|
|
OCL_TEST_P(FastNlMeansDenoising, Mat)
|
|
{
|
|
for (int j = 0; j < test_loop_times; j++)
|
|
{
|
|
generateTestData();
|
|
|
|
OCL_OFF(cv::fastNlMeansDenoising(src_roi, dst_roi, h, templateWindowSize, searchWindowSize));
|
|
OCL_ON(cv::fastNlMeansDenoising(usrc_roi, udst_roi, h, templateWindowSize, searchWindowSize));
|
|
|
|
OCL_EXPECT_MATS_NEAR(dst, 1);
|
|
}
|
|
}
|
|
|
|
typedef FastNlMeansDenoisingTestBase FastNlMeansDenoisingAbs;
|
|
|
|
OCL_TEST_P(FastNlMeansDenoisingAbs, Mat)
|
|
{
|
|
for (int j = 0; j < test_loop_times; j++)
|
|
{
|
|
generateTestData();
|
|
|
|
OCL_OFF(cv::fastNlMeansDenoisingAbs(src_roi, dst_roi, h, templateWindowSize, searchWindowSize));
|
|
OCL_ON(cv::fastNlMeansDenoisingAbs(usrc_roi, udst_roi, h, templateWindowSize, searchWindowSize));
|
|
|
|
OCL_EXPECT_MATS_NEAR(dst, 1);
|
|
}
|
|
}
|
|
|
|
typedef FastNlMeansDenoisingTestBase FastNlMeansDenoisingColored;
|
|
|
|
OCL_TEST_P(FastNlMeansDenoisingColored, Mat)
|
|
{
|
|
for (int j = 0; j < test_loop_times; j++)
|
|
{
|
|
generateTestData();
|
|
|
|
OCL_OFF(cv::fastNlMeansDenoisingColored(src_roi, dst_roi, h, h, templateWindowSize, searchWindowSize));
|
|
OCL_ON(cv::fastNlMeansDenoisingColored(usrc_roi, udst_roi, h, h, templateWindowSize, searchWindowSize));
|
|
|
|
OCL_EXPECT_MATS_NEAR(dst, 1);
|
|
}
|
|
}
|
|
|
|
OCL_INSTANTIATE_TEST_CASE_P(Photo, FastNlMeansDenoising,
|
|
Combine(Values(1, 2, 3, 4), Bool(), Bool()));
|
|
OCL_INSTANTIATE_TEST_CASE_P(Photo, FastNlMeansDenoisingAbs,
|
|
Combine(Values(1, 2, 3, 4), Bool(), Bool()));
|
|
OCL_INSTANTIATE_TEST_CASE_P(Photo, FastNlMeansDenoisingColored,
|
|
Combine(Values(3, 4), Bool(), Values(false)));
|
|
|
|
} } // namespace cvtest::ocl
|
|
|
|
#endif // HAVE_OPENCL
|