Gitea Mirror Bot
81 lines
2.1 KiB
C++
81 lines
2.1 KiB
C++
/*
|
|
* Copyright (c) 2024 Qualcomm Innovation Center, Inc. All rights reserved.
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
#include "test_precomp.hpp"
|
|
|
|
namespace opencv_test { namespace {
|
|
|
|
class DCTExtTest : public ::testing::TestWithParam<cv::Size> {};
|
|
|
|
TEST_P(DCTExtTest, forward)
|
|
{
|
|
Size size = GetParam();
|
|
|
|
RNG& rng = cv::theRNG();
|
|
Mat src(size, CV_8UC1);
|
|
cvtest::randUni(rng, src, Scalar::all(0), Scalar::all(255));
|
|
Mat srcFloat;
|
|
src.convertTo(srcFloat, CV_32F);
|
|
|
|
Mat dst, ref;
|
|
cv::fastcv::DCT(src, dst);
|
|
|
|
cv::dct(srcFloat, ref);
|
|
|
|
Mat dstFloat;
|
|
ref.convertTo(dstFloat, CV_32F);
|
|
|
|
double normInf = cvtest::norm(dstFloat, ref, cv::NORM_INF);
|
|
double normL2 = cvtest::norm(dstFloat, ref, cv::NORM_L2) / dst.size().area();
|
|
|
|
if (cvtest::debugLevel > 0)
|
|
{
|
|
std::cout << "dst:" << std::endl << dst << std::endl;
|
|
std::cout << "ref:" << std::endl << ref << std::endl;
|
|
}
|
|
|
|
EXPECT_EQ(normInf, 0);
|
|
EXPECT_EQ(normL2, 0);
|
|
}
|
|
|
|
TEST_P(DCTExtTest, inverse)
|
|
{
|
|
Size size = GetParam();
|
|
|
|
RNG& rng = cv::theRNG();
|
|
Mat src(size, CV_8UC1);
|
|
cvtest::randUni(rng, src, Scalar::all(0), Scalar::all(256));
|
|
|
|
Mat srcFloat;
|
|
src.convertTo(srcFloat, CV_32F);
|
|
|
|
Mat fwd, back;
|
|
cv::fastcv::DCT(src, fwd);
|
|
cv::fastcv::IDCT(fwd, back);
|
|
Mat backFloat;
|
|
back.convertTo(backFloat, CV_32F);
|
|
|
|
Mat fwdRef, backRef;
|
|
cv::dct(srcFloat, fwdRef);
|
|
cv::idct(fwdRef, backRef);
|
|
|
|
double normInf = cvtest::norm(backFloat, backRef, cv::NORM_INF);
|
|
double normL2 = cvtest::norm(backFloat, backRef, cv::NORM_L2) / src.size().area();
|
|
|
|
if (cvtest::debugLevel > 0)
|
|
{
|
|
std::cout << "src:" << std::endl << src << std::endl;
|
|
std::cout << "back:" << std::endl << back << std::endl;
|
|
std::cout << "backRef:" << std::endl << backRef << std::endl;
|
|
}
|
|
|
|
EXPECT_LE(normInf, 7.00005);
|
|
EXPECT_LT(normL2, 0.13);
|
|
}
|
|
|
|
INSTANTIATE_TEST_CASE_P(FastCV_Extension, DCTExtTest, ::testing::Values(Size(8, 8), Size(128, 128), Size(32, 256), Size(512, 512)));
|
|
|
|
}} // namespaces opencv_test, ::
|